Unverified 提交 f4b05680 authored 作者: Colin Wong's avatar Colin Wong 提交者: GitHub

Assert `--optimize` not used with cuda device (#8569)

上级 f8722b44
...@@ -492,6 +492,8 @@ def run( ...@@ -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)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论