yjh0410 1 년 전
부모
커밋
0b03030695
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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 = 0  # number of EMA updates
+        self.updates = 15510  # 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)