Unverified 提交 6b44ecd5 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

Follows #4823
上级 43b2817f
......@@ -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:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论