Fixed string format error during weight conversion (#1334)
This commit is contained in:
parent
e276e3a103
commit
8a414743e2
|
@ -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.')
|
||||||
|
|
Loading…
Reference in New Issue