Move check_amp()
to checks.py (#2948)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
@ -101,7 +101,7 @@ def test_val_scratch():
|
||||
|
||||
def test_amp():
|
||||
if torch.cuda.is_available():
|
||||
from ultralytics.yolo.engine.trainer import check_amp
|
||||
from ultralytics.yolo.utils.checks import check_amp
|
||||
model = YOLO(MODEL).model.cuda()
|
||||
assert check_amp(model)
|
||||
|
||||
|
Reference in New Issue
Block a user