Merge branch 'master' into instance_seg

AyushExel · Sep 11, 2022 · 6ddcf5a · 6ddcf5a
2 parents 3c84158 + 23701ea
commit 6ddcf5a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: