Edge TPU compiler comment (#6196)

* Edge TPU compiler comment

* 7 to 8 fix
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-01-04 19:36:12 -08:00
1 parent 9e9219fe17
commit 5402753a53
2 files changed
+5 -5

No files matched your search

+2 -2
View File
@@ -376,8 +376,8 @@ class DetectMultiBackend(nn.Module):
elif tflite: # https://www.tensorflow.org/lite/guide/python#install_tensorflow_lite_for_python
if 'edgetpu' in w.lower(): # Edge TPU
LOGGER.info(f'Loading {w} for TensorFlow Lite Edge TPU inference...')
import tflite_runtime.interpreter as tfli
delegate = {'Linux': 'libedgetpu.so.1', # install https://coral.ai/software/#edgetpu-runtime
import tflite_runtime.interpreter as tfli # install https://coral.ai/software/#edgetpu-runtime
delegate = {'Linux': 'libedgetpu.so.1',
'Darwin': 'libedgetpu.1.dylib',
'Windows': 'edgetpu.dll'}[platform.system()]
interpreter = tfli.Interpreter(model_path=w, experimental_delegates=[tfli.load_delegate(delegate)])