yjh0410 před 1 rokem
rodič
revize
100e554d3d
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      utils/misc.py

+ 1 - 1
utils/misc.py

@@ -378,7 +378,7 @@ class ModelEMA(object):
     def __init__(self, model, ema_decay=0.9999, ema_tau=2000, resume=None):
         # Create EMA
         self.ema = deepcopy(self.de_parallel(model)).eval()  # FP32 EMA
-        self.updates = 15510  # number of EMA updates
+        self.updates = 0  # number of EMA updates
         self.decay = lambda x: ema_decay * (1 - math.exp(-x / ema_tau))  # decay exponential ramp (to help early epochs)
         for p in self.ema.parameters():
             p.requires_grad_(False)