apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed string format error during weight conversion (#1334)

Browse Source
This commit is contained in:
Chang Lee
2020-06-22 22:07:51 -04:00
committed by GitHub
parent e276e3a103
commit 8a414743e2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models.py
+1 -1
View File
@@ -438,7 +438,7 @@ def convert(cfg='cfg/yolov3-spp.cfg', weights='weights/yolov3-spp.weights'):
target = weights.rsplit('.', 1)[0] + '.pt' target = weights.rsplit('.', 1)[0] + '.pt'
torch.save(chkpt, target) torch.save(chkpt, target)
print("Success: converted '%s' to 's%'" % (weights, target)) print("Success: converted '%s' to '%s'" % (weights, target))
else: else:
print('Error: extension not supported.') print('Error: extension not supported.')