From 46c55ac3bd9793127481a63220c99653ee529f42 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 17 Apr 2019 17:51:39 +0200
Subject: [PATCH] updates

---
 train.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/train.py b/train.py
index f487c7f3..d81de382 100644
--- a/train.py
+++ b/train.py
@@ -263,6 +263,10 @@ if __name__ == '__main__':
     opt = parser.parse_args()
     print(opt, end='\n\n')
 
+    if opt.evolve:
+        opt.notest = True  # save time by only testing final epoch
+        opt.nosave = True  # do not save checkpoints
+
     # Train
     results = train(
         opt.cfg,
@@ -279,7 +283,6 @@ if __name__ == '__main__':
 
     # Evolve hyperparameters (optional)
     if opt.evolve:
-        opt.notest = True  # save time by only testing final epoch
         best_fitness = results[2]  # use mAP for fitness
 
         gen = 30  # generations to evolve