Fix ema parameter

PaddlePaddle · Jun 13, 2023 · 29eb76b · 29eb76b
1 parent b4c0f12
commit 29eb76b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlers/tasks/object_detector.py b/paddlers/tasks/object_detector.py
@@ -447,7 +447,7 @@ def _real_train(
             self._is_backbone_weight())
 
         if use_ema:
-            ema = ModelEMA(model=self.net, decay=.9998, use_thres_step=True)
+            ema = ModelEMA(model=self.net, decay=.9998)
         else:
             ema = None
         # Start train loop