search

PixArt-alpha / PixArt-sigma

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation
https://pixart-alpha.github.io/PixArt-sigma-project/
GNU Affero General Public License v3.0
1.44k stars 67 forks source link

Make save_model_steps optional in train.py #97

Closed congruency closed 1 week ago

congruency commented 1 month ago

The epochs go very quickly with the smaller models. Saving by steps becomes unnecessary when an epoch completes in an hour on a smaller dataset.

This change modifies save_model logic to not perform when the config's save_model_steps is set to 0.

lawrence-cj commented 1 week ago

Great one. Thx man.