update: 适配昇腾

2025-11-27 09:57:34 +00:00 · 2025-11-27 09:57:34 +00:00 · 696c1b0793
commit 696c1b0793
parent e74b035c02
5 changed files with 34 additions and 11 deletions
--- a/ultralytics/engine/validator.py
+++ b/ultralytics/engine/validator.py
@ -158,7 +158,12 @@ class BaseValidator:
            self.dataloader = self.dataloader or self.get_dataloader(self.data.get(self.args.split), self.args.batch)

            model.eval()
-            model.warmup(imgsz=(1 if pt else self.args.batch, 3, imgsz, imgsz))  # warmup
+            # model.warmup(imgsz=(1 if pt else self.args.batch, 3, imgsz, imgsz))  # warmup
+            # 新增warmup，保证性能准确
+            print('start warm up')
+            model.warmup(imgsz=(self.args.batch, 3, imgsz, imgsz))  # warmup
+            model.warmup(imgsz=(self.args.batch, 3, 288, imgsz))  # warmup
+            print('end warm up')

        self.run_callbacks("on_val_start")
        dt = (