From 851c9b98839b18f867e37521d09a5688da90f095 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 14 Mar 2020 17:34:13 -0700
Subject: [PATCH] EMA class updates

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

diff --git a/train.py b/train.py
index a41a3179..83f51552 100644
--- a/train.py
+++ b/train.py
@@ -292,6 +292,7 @@ def train():
             if ni % accumulate == 0:
                 optimizer.step()
                 optimizer.zero_grad()
+                # ema.update(model)
 
             # Print batch results
             mloss = (mloss * i + loss_items) / (i + 1)  # update mean losses