updates
This commit is contained in:
parent
3ba7fc69b8
commit
fd3f2ed65f
2
train.py
2
train.py
|
@ -364,7 +364,7 @@ def train():
|
|||
# end epoch ----------------------------------------------------------------------------------------------------
|
||||
|
||||
# end training
|
||||
if len(opt.name):
|
||||
if len(opt.name) and not opt.prebias:
|
||||
os.rename('results.txt', 'results_%s.txt' % opt.name)
|
||||
os.rename(wdir + 'best.pt', wdir + 'best_%s.pt' % opt.name)
|
||||
os.rename(wdir + 'last.pt', wdir + 'last_%s.pt' % opt.name)
|
||||
|
|
Loading…
Reference in New Issue