Commit Graph
1 Commits
Author SHA1 Message Date
ZouJiu1andGlenn Jocher 2729761458 EMA changes for pre-model's batch_size (#3681)
* EMA changes for pre-model's batch_size

* Update train.py

* Update torch_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-18 16:26:52 +02:00