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 33232efac9fd..f5fb2c93a3d5 100755 --- a/utils/general.py +++ b/utils/general.py @@ -17,6 +17,7 @@ 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 amp_allclose(model, im): 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: