-
Notifications
You must be signed in to change notification settings - Fork 6.5k
expose polynomial:power and cosine_with_restarts:num_cycles params #1737
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
…_scheduler func, add it to train_dreambooth.py
|
The documentation is not available anymore as the PR was closed or merged. |
|
What's done:
Haven't found any test script in test folder that tests optimizers.py |
pcuenca
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good!
An alternative would be to use scheduler_kwargs in get_scheduler, but since we only have those two configurable schedulers I think it's good for now.
Co-authored-by: Pedro Cuenca <[email protected]>
patrickvonplaten
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @zetyquickly
…uggingface#1737) * expose polynomial:power and cosine_with_restarts:num_cycles using get_scheduler func, add it to train_dreambooth.py * fix formatting * fix style * Update src/diffusers/optimization.py Co-authored-by: Pedro Cuenca <[email protected]> Co-authored-by: Pedro Cuenca <[email protected]>
…uggingface#1737) * expose polynomial:power and cosine_with_restarts:num_cycles using get_scheduler func, add it to train_dreambooth.py * fix formatting * fix style * Update src/diffusers/optimization.py Co-authored-by: Pedro Cuenca <[email protected]> Co-authored-by: Pedro Cuenca <[email protected]>
to get_scheduler func,