diff --git a/train.py b/train.py index 4b0e8a5d..8f09bae3 100644 --- a/train.py +++ b/train.py @@ -437,8 +437,7 @@ if __name__ == '__main__': train() # train normally else: # Evolve hyperparameters (optional) - opt.notest = True # only test final epoch - opt.nosave = True # only save final checkpoint + opt.notest, opt.nosave = True, True # only test/save final epoch if opt.bucket: os.system('gsutil cp gs://%s/evolve.txt .' % opt.bucket) # download evolve.txt if exists