Преглед на файлове

Update

modifyDataloader
Glenn Jocher преди 2 години
родител
ревизия
779efbb9ca
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. +1
    -0
      utils/benchmarks.py

+ 1
- 0
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 == '-':

Loading…
Отказ
Запис