d8ahazard / sd_dreambooth_extension

Other
1.86k stars 281 forks source link

Option to stop train at max steps #1341

Closed Anime4000 closed 11 months ago

Anime4000 commented 1 year ago

Describe your changes

As per title, training will stop when Current Steps has been reached at Max Steps. this should act like old dreambooth where dataset x 100

Tested on 30 dataset, 4x4 (Batch * Grad) Default: 3200 steps @ 100 epoch Enabled: 3008 steps @ 94 epoch

image

The option is disabled by default, can be loaded via config

image

Show the state of Train until max steps in debug:

image

also fixed Sanity Sample Negative Prompt into the config, tired to paste every-time 🤣

Issue ticket number and link (if applicable)

Checklist before requesting a review