-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
change validation scheduler for train_dreambooth.py when training IF #4333
change validation scheduler for train_dreambooth.py when training IF #4333
Conversation
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. |
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.
Awesome!
if we want to make the scheduler configurable, I would prefer if the flag was to set a particular scheduler via a string instead of an adhoc override. Can you change the pr to pass a (pseudocode)
|
I use eval(args.validation_scheduler) to call DPMSolverMultistepScheduler. But the check reports: examples/dreambooth/train_dreambooth.py:50:5: F401 [*] |
Could we maybe work with importlib here instead of Python's |
…uggingface#4333) * dreambooth training * train_dreambooth validation scheduler * set a particular scheduler via a string * modify readme after setting a particular scheduler via a string * modify readme after setting a particular scheduler * use importlib to set a particular scheduler * import with correct sort
What does this PR do?
Fixes #4313
Who can review?
cc @sayakpaul and @williamberman for dreambooth