mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix wandb import issue

Browse Source
This commit is contained in:
James Betker 2021-06-04 23:27:15 -06:00
parent 103a88506e
commit fa908a6a15
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/train.py
View File

@ -270,6 +270,7 @@ class Trainer:
for ek, ev in eval_dict.items(): for ek, ev in eval_dict.items():
self.tb_logger.add_scalar(ek, ev, self.current_step) self.tb_logger.add_scalar(ek, ev, self.current_step)
if opt['wandb']: if opt['wandb']:
import wandb
wandb.log(eval_dict) wandb.log(eval_dict)