Commit
•
779efbb
1
Parent(s):
ad0e4d5
Update
Browse files- utils/benchmarks.py +1 -0
utils/benchmarks.py
CHANGED
@@ -58,6 +58,7 @@ def run(
|
|
58 |
device = select_device(device)
|
59 |
for i, (name, f, suffix, gpu) in formats.iterrows(): # index, (name, file, suffix, gpu-capable)
|
60 |
try:
|
|
|
61 |
if device.type != 'cpu':
|
62 |
assert gpu, f'{name} inference not supported on GPU'
|
63 |
if f == '-':
|
|
|
58 |
device = select_device(device)
|
59 |
for i, (name, f, suffix, gpu) in formats.iterrows(): # index, (name, file, suffix, gpu-capable)
|
60 |
try:
|
61 |
+
assert i < 9, 'Edge TPU and TF.js not supported'
|
62 |
if device.type != 'cpu':
|
63 |
assert gpu, f'{name} inference not supported on GPU'
|
64 |
if f == '-':
|