EMA changes for pre-model's batch_size (#3681) · fhkiel-mlaip/yolov5@72aa5b7 · GitHub

Commit

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

Browse files

* EMA changes for pre-model's batch_size

* Update train.py

* Update torch_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Loading branch information

ZouJiu1 and glenn-jocher authored Jun 18, 2021

1 parent 19b88b2 commit 72aa5b7