提交 412b113a authored 作者: Glenn Jocher's avatar Glenn Jocher

W&B logging bug fix (#1399)

上级 4250f84d
...@@ -221,7 +221,7 @@ def test(data, ...@@ -221,7 +221,7 @@ def test(data,
nt = torch.zeros(1) nt = torch.zeros(1)
# W&B logging # W&B logging
if plots and wandb: if plots and wandb and wandb.run:
wandb.log({"Images": wandb_images}) wandb.log({"Images": wandb_images})
wandb.log({"Validation": [wandb.Image(str(x), caption=x.name) for x in sorted(save_dir.glob('test*.jpg'))]}) wandb.log({"Validation": [wandb.Image(str(x), caption=x.name) for x in sorted(save_dir.glob('test*.jpg'))]})
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论