From 2729761458cdf868af3dad496a0dfcb2fd1d5aa4 Mon Sep 17 00:00:00 2001
From: ZouJiu1 <34758215+ZouJiu1@users.noreply.github.com>
Date: Fri, 18 Jun 2021 22:26:52 +0800
Subject: [PATCH] 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>