Kohya-ss-sd-scripts/train_db.py at db845300748a600bb5857852deebac1d5f9d3a7f

mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00

Files

Isotr0py db84530074 Fix gradients synchronization for multi-GPUs training (#989 )

* delete DDP wrapper

* fix train_db vae and train_network

* fix train_db vae and train_network unwrap

* network grad sync

---------

Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com>

2023-12-07 22:01:42 +09:00

22 KiB

Raw Blame History

View Raw

22 KiB Raw Blame History

22 KiB

Raw Blame History