Bläddra i källkod

Fix ema parameter

Bobholamovic 2 år sedan
förälder
incheckning
29eb76bbaf
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1 1
      paddlers/tasks/object_detector.py

+ 1 - 1
paddlers/tasks/object_detector.py

@@ -447,7 +447,7 @@ class BaseDetector(BaseModel):
             self._is_backbone_weight())
             self._is_backbone_weight())
 
 
         if use_ema:
         if use_ema:
-            ema = ModelEMA(model=self.net, decay=.9998, use_thres_step=True)
+            ema = ModelEMA(model=self.net, decay=.9998)
         else:
         else:
             ema = None
             ema = None
         # Start train loop
         # Start train loop