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

Log eval to wandb

Browse Source
This commit is contained in:
James Betker 2021-06-04 23:23:20 -06:00
parent 7d45132f60
commit 103a88506e
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/requirements.txt
View File

@ -19,3 +19,4 @@ orjson
einops einops
gsa-pytorch gsa-pytorch
lambda-networks lambda-networks
pytorch_ssim

3
codes/train.py
View File

@ -269,6 +269,9 @@ class Trainer:
print("Evaluator results: ", eval_dict) print("Evaluator results: ", eval_dict)
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']:
wandb.log(eval_dict)
def do_training(self): def do_training(self):
self.logger.info('Start training from epoch: {:d}, iter: {:d}'.format(self.start_epoch, self.current_step)) self.logger.info('Start training from epoch: {:d}, iter: {:d}'.format(self.start_epoch, self.current_step))