瀏覽代碼
Fix 'PyTorch starting from' for URL weights (#4828)
Follows #4823
modifyDataloader
Glenn Jocher
GitHub
3 年之前
沒有發現已知的金鑰在資料庫的簽署中
GPG Key ID: 4AEE18F83AFDEB23
共有
1 個文件被更改,包括
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: |