From 23701eac7a7b160e478ba4bbef966d0af9348251 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 11 Sep 2022 13:56:51 +0300 Subject: [PATCH] AMP Check fix (#9367) Resolves https://github.com/ultralytics/yolov5/issues/9365 Signed-off-by: Glenn Jocher Signed-off-by: Glenn Jocher --- utils/general.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils/general.py b/utils/general.py index 33232ef..f5fb2c9 100755 --- a/utils/general.py +++ b/utils/general.py @@ -17,6 +17,7 @@ import signal import sys import time import urllib +from copy import deepcopy from datetime import datetime from itertools import repeat from multiprocessing.pool import ThreadPool @@ -535,7 +536,7 @@ def check_amp(model): f = ROOT / 'data' / 'images' / 'bus.jpg' # image to check im = f if f.exists() else 'https://ultralytics.com/images/bus.jpg' if check_online() else np.ones((640, 640, 3)) try: - assert amp_allclose(model, im) or amp_allclose(DetectMultiBackend('yolov5n.pt', device), im) + assert amp_allclose(deepcopy(model), im) or amp_allclose(DetectMultiBackend('yolov5n.pt', device), im) LOGGER.info(f'{prefix}checks passed ✅') return True except Exception: