From 779efbb9ca26b9ed4177a59936ec1d0dfdc9365e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 3 Apr 2022 21:21:55 +0200 Subject: [PATCH] Update --- utils/benchmarks.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/benchmarks.py b/utils/benchmarks.py index 6924372..36e8278 100644 --- 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 == '-':