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

Daemon thread mosaic plots fix (#1943)

上级 3a56cac4
...@@ -318,8 +318,9 @@ def train(hyp, opt, device, tb_writer=None, wandb=None): ...@@ -318,8 +318,9 @@ 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 > 3 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()]})
# end batch ------------------------------------------------------------------------------------------------ # end batch ------------------------------------------------------------------------------------------------
# end epoch ---------------------------------------------------------------------------------------------------- # end epoch ----------------------------------------------------------------------------------------------------
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论