updates

2020-01-30 14:32:10 -08:00 · 2020-01-30 14:32:10 -08:00 · 4b9d73f931
parent ac8d78382a
commit 4b9d73f931
1 changed files with 1 additions and 2 deletions
--- 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