Glenn Jocher
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
train.py
View File
| # 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 == 10 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()]}) | if x.exists()]}) | ||||
Loading…