Log eval to wandb

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

View File

@ -18,4 +18,5 @@ vector_quantize_pytorch
orjson
einops
gsa-pytorch
lambda-networks
lambda-networks
pytorch_ssim

View File

@ -269,6 +269,9 @@ class Trainer:
print("Evaluator results: ", eval_dict)
for ek, ev in eval_dict.items():
self.tb_logger.add_scalar(ek, ev, self.current_step)
if opt['wandb']:
wandb.log(eval_dict)
def do_training(self):
self.logger.info('Start training from epoch: {:d}, iter: {:d}'.format(self.start_epoch, self.current_step))