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:
1 parent
f527704cd3
commit
2729761458