Unverified 提交 f4a78e1b authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

W&B mosaic log bug fix (#1949)

上级 03ebe6ea
...@@ -318,7 +318,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None): ...@@ -318,7 +318,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
# if tb_writer: # if tb_writer:
# tb_writer.add_image(f, result, dataformats='HWC', global_step=epoch) # tb_writer.add_image(f, result, dataformats='HWC', global_step=epoch)
# tb_writer.add_graph(model, imgs) # add model to tensorboard # 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') wandb.log({"Mosaics": [wandb.Image(str(x), caption=x.name) for x in save_dir.glob('train*.jpg')
if x.exists()]}) if x.exists()]})
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论