Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

W&B logging bug fix (#1399)

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-11-17 19:17:57 +01:00
1 parent 4250f84dfb
commit 412b113ac6
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
test.py
+1 -1
View File
@@ -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'))]})