diff --git a/model/pytorch/dcrnn_supervisor.py b/model/pytorch/dcrnn_supervisor.py index c09052e..0b36519 100644 --- a/model/pytorch/dcrnn_supervisor.py +++ b/model/pytorch/dcrnn_supervisor.py @@ -133,7 +133,7 @@ class DCRNNSupervisor: per_timestep_loss /= num_batches for i, val in enumerate(per_timestep_loss): - self._logger.info("Dataset:{}, Timestep: {}, MAE:{}".format(dataset, i, val.item())) + self._logger.info("Dataset:{}, Timestep: {}, MAE:{:.4f}".format(dataset, i, val.item())) return mean_loss