田运杰 10 månader sedan
förälder
incheckning
4d00e60c70
1 ändrade filer med 2 tillägg och 1 borttagningar
  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(