This commit is contained in:
Glenn Jocher committed 2022-04-03 21:21:55 +02:00
1 parent ad0e4d5d19
commit 779efbb9ca
1 file changed
+1
+1
View File
@@ -58,6 +58,7 @@ def run(
device = select_device(device)
for i, (name, f, suffix, gpu) in formats.iterrows(): # index, (name, file, suffix, gpu-capable)
try:
assert i < 9, 'Edge TPU and TF.js not supported'
if device.type != 'cpu':
assert gpu, f'{name} inference not supported on GPU'
if f == '-':