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