Fix wandb import issue

2021-06-04 23:27:15 -06:00 · 2021-06-04 23:27:15 -06:00 · fa908a6a15
commit fa908a6a15
parent 103a88506e
1 changed files with 1 additions and 0 deletions
--- a/codes/train.py
+++ b/codes/train.py
@ -270,6 +270,7 @@ class Trainer:
                for ek, ev in eval_dict.items():
                    self.tb_logger.add_scalar(ek, ev, self.current_step)
                if opt['wandb']:
+                    import wandb
                    wandb.log(eval_dict)