examples/custom_diffusion/train_custom_diffusion.py · ece55227ffc60f7bb09442b390a02e45ac0438f8 · chenpangpang / diffusers

"...resnet50_tensorflow.git" did not exist on "fd232c2c79cbc45d5ba2683ca1f5568763ade254"

Ruoxi authored Jul 13, 2023

* Multiply lr scheduler steps by `num_processes`.

* Stop multiplying steps by gradient accumulation.

ece55227

train_custom_diffusion.py 56.2 KB