device = select_device(device) | device = select_device(device) | ||||
for i, (name, f, suffix, gpu) in formats.iterrows(): # index, (name, file, suffix, gpu-capable) | for i, (name, f, suffix, gpu) in formats.iterrows(): # index, (name, file, suffix, gpu-capable) | ||||
try: | try: | ||||
assert i < 9, 'Edge TPU and TF.js not supported' | |||||
if device.type != 'cpu': | if device.type != 'cpu': | ||||
assert gpu, f'{name} inference not supported on GPU' | assert gpu, f'{name} inference not supported on GPU' | ||||
if f == '-': | if f == '-': |