Fix 'PyTorch starting from' for URL weights (#4828)
Browse files
export.py
CHANGED
@@ -272,7 +272,7 @@ def run(data=ROOT / 'data/coco128.yaml', # 'dataset.yaml path'
|
|
272 |
|
273 |
for _ in range(2):
|
274 |
y = model(im) # dry runs
|
275 |
-
print(f"\n{colorstr('PyTorch:')} starting from {
|
276 |
|
277 |
# Exports
|
278 |
if 'torchscript' in include:
|
|
|
272 |
|
273 |
for _ in range(2):
|
274 |
y = model(im) # dry runs
|
275 |
+
print(f"\n{colorstr('PyTorch:')} starting from {file} ({file_size(file):.1f} MB)")
|
276 |
|
277 |
# Exports
|
278 |
if 'torchscript' in include:
|