소스 검색

W&B logging bug fix (#1399)

5.0
Glenn Jocher 3 년 전
부모
커밋
412b113ac6
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. +1
    -1
      test.py

+ 1
- 1
test.py 파일 보기

@@ -221,7 +221,7 @@ def test(data,
nt = torch.zeros(1)

# W&B logging
if plots and wandb:
if plots and wandb and wandb.run:
wandb.log({"Images": wandb_images})
wandb.log({"Validation": [wandb.Image(str(x), caption=x.name) for x in sorted(save_dir.glob('test*.jpg'))]})


Loading…
취소
저장