From 94e1749a1dfb9a5e9337a0540005ebbdeb9bd763 Mon Sep 17 00:00:00 2001 From: phlrain Date: Tue, 19 Nov 2024 22:01:40 +0800 Subject: [PATCH] update --- ppdet/engine/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppdet/engine/trainer.py b/ppdet/engine/trainer.py index 4c45e302..c36607e2 100644 --- a/ppdet/engine/trainer.py +++ b/ppdet/engine/trainer.py @@ -429,8 +429,8 @@ def train(self, validate=False): model = self.model if self.cfg.get('to_static', False): model = apply_to_static(self.cfg, model) - if self.cfg.architecture == 'YOLOv5': - model.yolo_head.loss.to_static = True + #if self.cfg.architecture == 'YOLOv5': + # model.yolo_head.loss.to_static = True sync_bn = ( getattr(self.cfg, 'norm_type', None) == 'sync_bn' and (self.cfg.use_gpu or self.cfg.use_npu or self.cfg.use_mlu) and