提交 779efbb9 authored 作者: Glenn Jocher's avatar Glenn Jocher

Update

上级 ad0e4d5d
...@@ -58,6 +58,7 @@ def run( ...@@ -58,6 +58,7 @@ def run(
device = select_device(device) device = select_device(device)
for i, (name, f, suffix, gpu) in formats.iterrows(): # index, (name, file, suffix, gpu-capable) for i, (name, f, suffix, gpu) in formats.iterrows(): # index, (name, file, suffix, gpu-capable)
try: try:
assert i < 9, 'Edge TPU and TF.js not supported'
if device.type != 'cpu': if device.type != 'cpu':
assert gpu, f'{name} inference not supported on GPU' assert gpu, f'{name} inference not supported on GPU'
if f == '-': if f == '-':
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论