Add hparams metrics to tensorboard
Currently, we track hparams, but not metrics which is a new feature.
https://pytorch.org/docs/stable/tensorboard.html#torch.utils.tensorboard.writer.SummaryWriter.add_hparams