From 33264f55679560284900f76c385fa54a7e0f86b3 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 12 Jan 2020 16:18:29 -0800 Subject: [PATCH] updates --- train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index 2ec9db9b..44ac3708 100644 --- a/train.py +++ b/train.py @@ -383,7 +383,8 @@ def train(): if opt.bucket: os.system('gsutil cp %s %s gs://%s' % (fresults, wdir + flast, opt.bucket)) - plot_results() # save as results.png + if not opt.evolve: + 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 torch.cuda.empty_cache()