ZouJiu1 glenn-jocher commited on
Commit
2729761
·
unverified ·
1 Parent(s): f527704

EMA changes for pre-model's batch_size (#3681)

Browse files

* EMA changes for pre-model's batch_size

* Update train.py

* Update torch_utils.py

Co-authored-by: Glenn Jocher <[email protected]>

Files changed (0) hide show