From facce8a0c9f28736104fa89a1fe76147127cf262 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 16 Sep 2021 14:38:35 +0200 Subject: [PATCH] Fix 'PyTorch starting from' for URL weights (#4828) Follows #4823 --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 4ec3c3e0c711..a5139c0a965e 100644 --- a/export.py +++ b/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: