W&B mosaic log bug fix (ultralytics#1949)

fhkiel-mlaip · Jan 15, 2021 · 47ed153 · 47ed153
1 parent 40b1978
commit 47ed153
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -318,7 +318,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
                     # if tb_writer:
                     #     tb_writer.add_image(f, result, dataformats='HWC', global_step=epoch)
                     #     tb_writer.add_graph(model, imgs)  # add model to tensorboard
-                elif plots and ni > 3 and wandb:
+                elif plots and ni == 10 and wandb:
                     wandb.log({"Mosaics": [wandb.Image(str(x), caption=x.name) for x in save_dir.glob('train*.jpg')
                                            if x.exists()]})