glenn-jocher
commited on
Commit
•
f4a78e1
1
Parent(s):
03ebe6e
W&B mosaic log bug fix (#1949)
Browse files
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
|
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 |
|