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