Update

fhkiel-mlaip · Apr 3, 2022 · 73ace7a · 73ace7a
1 parent 044825d
commit 73ace7a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/benchmarks.py b/utils/benchmarks.py
@@ -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 == '-':