Переглянути джерело

Fix 'PyTorch starting from' for URL weights (#4828)

Follows #4823
modifyDataloader
Glenn Jocher GitHub 3 роки тому
джерело
коміт
6b44ecd53d
Не вдалося знайти GPG ключ що відповідає даному підпису Ідентифікатор GPG ключа: 4AEE18F83AFDEB23
1 змінених файлів з 1 додано та 1 видалено
  1. +1
    -1
      export.py

+ 1
- 1
export.py Переглянути файл

@@ -272,7 +272,7 @@ def run(data=ROOT / 'data/coco128.yaml', # 'dataset.yaml path'

for _ in range(2):
y = model(im) # dry runs
print(f"\n{colorstr('PyTorch:')} starting from {weights} ({file_size(weights):.1f} MB)")
print(f"\n{colorstr('PyTorch:')} starting from {file} ({file_size(file):.1f} MB)")

# Exports
if 'torchscript' in include:

Завантаження…
Відмінити
Зберегти