From 29eb76bbafc33c71d50cace17a6651f95fd131af Mon Sep 17 00:00:00 2001 From: Bobholamovic Date: Tue, 13 Jun 2023 11:47:01 +0800 Subject: [PATCH] Fix ema parameter --- paddlers/tasks/object_detector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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