From 6ccf19038de2bb6bef9b69ce5fa1fa180ec7e2c9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 21 Jan 2020 16:18:24 -0800 Subject: [PATCH] updates --- train.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index 73b08ac4..2249c45a 100644 --- a/train.py +++ b/train.py @@ -365,8 +365,8 @@ def train(): torch.save(chkpt, best) # Save backup every 10 epochs (optional) - if epoch > 0 and epoch % 10 == 0: - torch.save(chkpt, wdir + 'backup%g.pt' % epoch) + # if epoch > 0 and epoch % 10 == 0: + # torch.save(chkpt, wdir + 'backup%g.pt' % epoch) # Delete checkpoint del chkpt @@ -384,7 +384,8 @@ def train(): # save to cloud if opt.bucket: - os.system('gsutil cp %s %s gs://%s' % (fresults, wdir + flast, opt.bucket)) + os.system('gsutil cp %s gs://%s/results' % (fresults, opt.bucket)) + os.system('gsutil cp %s gs://%s/weights' % (wdir + flast, opt.bucket)) if not opt.evolve: plot_results() # save as results.png