From fb81559565bb6fd2c693cfe73c2718159d5baf63 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 11 Sep 2019 23:11:01 +0200
Subject: [PATCH] updates

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index fb3d0e4c..a461942d 100644
--- a/train.py
+++ b/train.py
@@ -327,7 +327,7 @@ def train():
             best_fitness = fitness
 
         # Save training results
-        save = (not opt.nosave) or ((not opt.evolve) and final_epoch)
+        save = (not opt.nosave) or (final_epoch and not opt.evolve) or opt.prebias
         if save:
             with open('results.txt', 'r') as file:
                 # Create checkpoint
@@ -340,7 +340,7 @@ def train():
 
             # Save last checkpoint
             torch.save(chkpt, last)
-            if opt.bucket and not opt.evolve:
+            if opt.bucket and not opt.prebias:
                 os.system('gsutil cp %s gs://%s' % (last, opt.bucket))  # upload to bucket
 
             # Save best checkpoint