From 3ba7fc69b871694a09ada8bb59e2f26370ecffa3 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 2 Nov 2019 19:47:25 -0700 Subject: [PATCH] updates --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index 7ada1dfc..26dd2d60 100644 --- a/train.py +++ b/train.py @@ -367,6 +367,7 @@ def train(): if len(opt.name): 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) 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