glenn-jocher commited on
Commit
779efbb
1 Parent(s): ad0e4d5
Files changed (1) hide show
  1. 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 == '-':