Add IVON optimizer support #2129
rockerBOO
wants to merge
8 commits into
kohya-ss:sd3
from
rockerBOO:ivon
Add IVON optimizer support
7ef68b5d
Add tests and pruning
8cdfb202
Add --enable_pruning option
30f479fa
Remove random pruning
47a0a9fa
Add sample params test for IVON
6b810499
Add IVON to tests
ba467e61
Fix pruning
ee922596
Revert zero_grad
ee282be9
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub