Fix results_file not renaming (#903)
Browse files
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
|
367 |
-
for f1, f2 in zip([wdir / 'last.pt', wdir / 'best.pt',
|
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
|