diffusers
de45af4a - Allow setting num_cycles for cosine_with_restarts lr scheduler (#3606)

Commit

2 years ago

Allow setting num_cycles for cosine_with_restarts lr scheduler (#3606) Expose num_cycles kwarg of get_schedule() through args.lr_num_cycles.

References

Author

0x1355

Parents