Glenn Jocher
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
test.py
View File
| 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'))]}) | ||||
Loading…