Unverified 提交 d8274d04 authored 作者: NanoCode012's avatar NanoCode012 提交者: GitHub

Fix results_file not renaming (#903)

上级 281d78c1
...@@ -363,8 +363,8 @@ def train(hyp, opt, device, tb_writer=None): ...@@ -363,8 +363,8 @@ def train(hyp, opt, device, tb_writer=None):
if rank in [-1, 0]: if rank in [-1, 0]:
# Strip optimizers # Strip optimizers
n = opt.name if opt.name.isnumeric() else '' n = opt.name if opt.name.isnumeric() else ''
fresults, flast, fbest = 'results%s.txt' % n, wdir / f'last{n}.pt', wdir / f'best{n}.pt' fresults, flast, fbest = log_dir / f'results{n}.txt', wdir / f'last{n}.pt', wdir / f'best{n}.pt'
for f1, f2 in zip([wdir / 'last.pt', wdir / 'best.pt', 'results.txt'], [flast, fbest, fresults]): for f1, f2 in zip([wdir / 'last.pt', wdir / 'best.pt', results_file], [flast, fbest, fresults]):
if os.path.exists(f1): if os.path.exists(f1):
os.rename(f1, f2) # rename os.rename(f1, f2) # rename
if str(f2).endswith('.pt'): # is *.pt if str(f2).endswith('.pt'): # is *.pt
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论