From 6b44ecd53dd299ccaa54cff4194e0f4e323bbc40 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
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 4ec3c3e..a5139c0 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: