updates

2019-10-13 18:53:15 +02:00 · 2019-10-13 18:53:15 +02:00 · 376e00a3cf
parent 139161c522
commit 376e00a3cf
1 changed files with 1 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -366,6 +366,7 @@ def train():
    # end training
    if len(opt.name):
        os.rename('results.txt', 'results_%s.txt' % opt.name)
+        os.rename(wdir + 'best.pt', wdir + 'best_%s.pt' % opt.name)
    plot_results()  # save as results.png
    print('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600))
    dist.destroy_process_group() if torch.cuda.device_count() > 1 else None