From 41bf46a4191cf04bfeda78a3e82bc24c2cd6ffb0 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 11 Mar 2020 22:11:19 -0700
Subject: [PATCH] updates

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index d35407fb..36cef234 100644
--- a/train.py
+++ b/train.py
@@ -240,7 +240,7 @@ def train():
             # Hyperparameter Burn-in
             n_burn = 200  # number of burn-in batches
             if ni <= n_burn:
-                g = (ni / n_burn) ** 2  # gain
+                g = ni / n_burn  # gain
                 for x in model.named_modules():
                     if x[0].endswith('BatchNorm2d'):
                         # x[1].momentum = 1 - 0.9 * g  # momentum falls from 1 - 0.1