glenn-jocher commited on
Commit
6b44ecd
·
unverified ·
1 Parent(s): 43b2817

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

Browse files
Files changed (1) hide show
  1. export.py +1 -1
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 {weights} ({file_size(weights):.1f} MB)")
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: