Fix device-id parameter in edgetpu_benchmark.py

markus-k · May 3, 2021 · e018b3b · e018b3b
1 parent 6914025
commit e018b3b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/edgetpu_benchmark.py b/edgetpu_benchmark.py
@@ -26,9 +26,11 @@
 DEVICE_ID = args.device_id
 USE_EDGETPU = (DEVICE != 'cpu')
 
-if DEVICE_ID:
+if DEVICE_ID is not None:
     DEVICE = f'{DEVICE}:{DEVICE_ID}'
 
+print(f'Using device: {DEVICE}')
+
 model_path = args.model
 
 COUNT = args.count