From 5d7694e293c37faa5ab30323b44510b14c7d7f3e Mon Sep 17 00:00:00 2001 From: Chintan Shah Date: Mon, 7 Oct 2019 20:54:58 -0400 Subject: [PATCH] logging to 4 decimals --- model/pytorch/dcrnn_supervisor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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