|
@@ -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
|