examples/instruct_pix2pix/train_instruct_pix2pix.py · b28675c605a89c3da94e9792bec91b32209d191b · renzhc / diffusers_dcu

[train_instruct_pix2pix.py]Fix the LR schedulers when `num_train_epochs` is... · b28675c6

Anand Kumar authored Sep 28, 2024


[train_instruct_pix2pix.py]Fix the LR schedulers when `num_train_epochs` is passed in a distributed training env (#9316)

Fixed pix2pix lr scheduler
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

b28675c6

train_instruct_pix2pix.py 43.6 KB

Replace train_instruct_pix2pix.py