From ce3872f86c2fc90698e84aed66091a5b5cb8952b Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 12 Nov 2020 12:55:52 +0100
Subject: [PATCH] W&B evolution bug fix (#1373)

---
 train.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 0dce414..bfa1e03 100644
--- a/train.py
+++ b/train.py
@@ -323,7 +323,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
                                                  dataloader=testloader,
                                                  save_dir=log_dir,
                                                  plots=epoch == 0 or final_epoch,  # plot first and last
-                                                 log_imgs=opt.log_imgs)
+                                                 log_imgs=opt.log_imgs if wandb else 0)
 
             # Write
             with open(results_file, 'a') as f:
@@ -475,7 +475,6 @@ if __name__ == '__main__':
 
                 assert os.environ.get('WANDB_DISABLED') != 'true'
             except (ImportError, AssertionError):
-                opt.log_imgs = 0
                 logger.info("Install Weights & Biases for experiment logging via 'pip install wandb' (recommended)")
 
         train(hyp, opt, device, tb_writer, wandb)