田运杰 11 tháng trước cách đây
mục cha
commit
4d00e60c70
1 tập tin đã thay đổi với 2 bổ sung1 xóa
  1. 2 1
      ultralytics/utils/checks.py

+ 2 - 1
ultralytics/utils/checks.py

@@ -702,7 +702,8 @@ def check_amp(model):
     try:
         from ultralytics import YOLO
 
-        assert amp_allclose(YOLO("yolo11n.pt"), im)
+        # assert amp_allclose(YOLO("yolo11n.pt"), im)
+        assert amp_allclose(YOLO("yolov12n.pt"), im)
         LOGGER.info(f"{prefix}checks passed ✅")
     except ConnectionError:
         LOGGER.warning(