glenn-jocher commited on
Commit
f4a78e1
1 Parent(s): 03ebe6e

W&B mosaic log bug fix (#1949)

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -318,7 +318,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
318
  # if tb_writer:
319
  # tb_writer.add_image(f, result, dataformats='HWC', global_step=epoch)
320
  # tb_writer.add_graph(model, imgs) # add model to tensorboard
321
- elif plots and ni > 3 and wandb:
322
  wandb.log({"Mosaics": [wandb.Image(str(x), caption=x.name) for x in save_dir.glob('train*.jpg')
323
  if x.exists()]})
324
 
 
318
  # if tb_writer:
319
  # tb_writer.add_image(f, result, dataformats='HWC', global_step=epoch)
320
  # tb_writer.add_graph(model, imgs) # add model to tensorboard
321
+ elif plots and ni == 10 and wandb:
322
  wandb.log({"Mosaics": [wandb.Image(str(x), caption=x.name) for x in save_dir.glob('train*.jpg')
323
  if x.exists()]})
324