Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

AMP Check fix (#9367)

Browse Source 
Resolves https://github.com/ultralytics/yolov5/issues/9365

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-09-11 13:56:51 +03:00
1 parent 23d0456b08
commit 23701eac7a
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
utils/general.py
+2 -1
View File
@@ -17,6 +17,7 @@ import signal
import sys import sys
import time import time
import urllib import urllib
from copy import deepcopy
from datetime import datetime from datetime import datetime
from itertools import repeat from itertools import repeat
from multiprocessing.pool import ThreadPool from multiprocessing.pool import ThreadPool
@@ -535,7 +536,7 @@ def check_amp(model):
f = ROOT / 'data' / 'images' / 'bus.jpg' # image to check 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)) im = f if f.exists() else 'https://ultralytics.com/images/bus.jpg' if check_online() else np.ones((640, 640, 3))
try: 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 ✅') LOGGER.info(f'{prefix}checks passed ✅')
return True return True
except Exception: except Exception: