Nanobit commited on
Commit
d8274d0
·
unverified ·
1 Parent(s): 281d78c

Fix results_file not renaming (#903)

Browse files
Files changed (1) hide show
  1. train.py +2 -2
train.py CHANGED
@@ -363,8 +363,8 @@ def train(hyp, opt, device, tb_writer=None):
363
  if rank in [-1, 0]:
364
  # Strip optimizers
365
  n = opt.name if opt.name.isnumeric() else ''
366
- fresults, flast, fbest = 'results%s.txt' % n, wdir / f'last{n}.pt', wdir / f'best{n}.pt'
367
- for f1, f2 in zip([wdir / 'last.pt', wdir / 'best.pt', 'results.txt'], [flast, fbest, fresults]):
368
  if os.path.exists(f1):
369
  os.rename(f1, f2) # rename
370
  if str(f2).endswith('.pt'): # is *.pt
 
363
  if rank in [-1, 0]:
364
  # Strip optimizers
365
  n = opt.name if opt.name.isnumeric() else ''
366
+ fresults, flast, fbest = log_dir / f'results{n}.txt', wdir / f'last{n}.pt', wdir / f'best{n}.pt'
367
+ for f1, f2 in zip([wdir / 'last.pt', wdir / 'best.pt', results_file], [flast, fbest, fresults]):
368
  if os.path.exists(f1):
369
  os.rename(f1, f2) # rename
370
  if str(f2).endswith('.pt'): # is *.pt