Nerogar / OneTrainer

OneTrainer is a one-stop solution for all your stable diffusion training needs.
GNU Affero General Public License v3.0
1.81k stars 154 forks source link

[Feat]: Update schedulefree to 1.3 #580

Closed clayne closed 1 week ago

clayne commented 1 week ago

Describe your use-case.

Both of these issues were due to a bug in schedulefree 1.2.7:

https://github.com/Nerogar/OneTrainer/issues/578 https://github.com/Nerogar/OneTrainer/issues/579 (although this one seems potentially less related)

and discussed here: https://discord.com/channels/1102003518203756564/1102013112124706827/1308522446575632388

I've checked the changelog for schedulefree 1.3 and the one major change is related to weight decay during learning rate warmup: https://github.com/facebookresearch/schedule_free?tab=readme-ov-file#releases

Other than that, everything else looks minor.

What would you like to see as a solution?

From:

requirements-global.txt:schedulefree==1.2.7 # schedule-free optimizers

To:

requirements-global.txt:schedulefree==1.3 # schedule-free optimizers

Have you considered alternatives? List them here.

No response