diff --git a/paddlers/tasks/object_detector.py b/paddlers/tasks/object_detector.py index 3d64ad12..92fb8c0c 100644 --- 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