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

Follows ultralytics#4823
fhkiel-mlaip · Sep 16, 2021 · 18d2cc2 · 18d2cc2
1 parent a0665ef
commit 18d2cc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: