From 985006a52ab9d4f27c5bf66a9c28dc9298835593 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 14 Nov 2019 17:25:29 -0800 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 5f61f7c4..ec913098 100644 --- a/train.py +++ b/train.py @@ -366,7 +366,7 @@ def train(): # end training if len(opt.name) and not opt.prebias: os.rename('results.txt', 'results_%s.txt' % opt.name) - os.rename(wdir + 'last.pt', wdir + 'last_%s.pt' % opt.name) + os.rename(wdir + 'last.pt', wdir + 'last_%s.pt' % opt.name) if os.path.exists(wdir + 'last.pt') else None os.rename(wdir + 'best.pt', wdir + 'best_%s.pt' % opt.name) if os.path.exists(wdir + 'best.pt') else None plot_results() # save as results.png print('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600))