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>
This commit is contained in:
ZouJiu1andGlenn Jocher authored and GitHub committed 2021-06-18 16:26:52 +02:00
1 parent f527704cd3
commit 2729761458

Diff Content Not Available