diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard

expose polynomial:power and cosine_with_restarts:num_cycles params

Open zetyquickly opened this issue 3 years ago • 2 comments

to get_scheduler func,

  • add it to the train_dreambooth.py

zetyquickly avatar Dec 17 '22 07:12 zetyquickly

The documentation is not available anymore as the PR was closed or merged.

@pcuenca PR to an issue https://github.com/huggingface/diffusers/issues/1232

What's done:

  • exposed only two parameters power and num_cycles on top of existing get_scheduler func API. it's possible to expose all of the parameters but taking into account that the issue wasn't there before we've noticed it, I assume it would pile up the code
  • changed parameter num_cycles name of the cosine scheduler to not overlap with cosine_with_restarts, because they have different meanings
  • added new parameters to the dreambooth's args

Haven't found any test script in test folder that tests optimizers.py

zetyquickly avatar Dec 17 '22 07:12 zetyquickly