forked from mrq/DL-Art-School
Log eval to wandb
This commit is contained in:
parent
7d45132f60
commit
103a88506e
|
@ -18,4 +18,5 @@ vector_quantize_pytorch
|
|||
orjson
|
||||
einops
|
||||
gsa-pytorch
|
||||
lambda-networks
|
||||
lambda-networks
|
||||
pytorch_ssim
|
|
@ -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))
|
||||
|
|
Loading…
Reference in New Issue
Block a user