Assert --optimize not used with cuda device (#8569)
This commit is contained in:
1 parent
f8722b4429
commit
f4b05680f8
1 file changed
+2
@@ -492,6 +492,8 @@ def run(
|
|||||||
# Checks
|
# Checks
|
||||||
imgsz *= 2 if len(imgsz) == 1 else 1 # expand
|
imgsz *= 2 if len(imgsz) == 1 else 1 # expand
|
||||||
assert nc == len(names), f'Model class count {nc} != len(names) {len(names)}'
|
assert nc == len(names), f'Model class count {nc} != len(names) {len(names)}'
|
||||||
|
if optimize:
|
||||||
|
assert device.type != 'cuda', '--optimize not compatible with cuda devices, i.e. use --device cpu'
|
||||||
|
|
||||||
# Input
|
# Input
|
||||||
gs = int(max(model.stride)) # grid size (max stride)
|
gs = int(max(model.stride)) # grid size (max stride)
|
||||||
|
|||||||
Reference in new issue
Block a user