From 036e552bf6c4df64307751165350e22a06819b03 Mon Sep 17 00:00:00 2001 From: Chintan Shah Date: Sun, 6 Oct 2019 14:31:46 -0400 Subject: [PATCH] logging and refactor --- 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 592454b..84d0938 100644 --- a/model/pytorch/dcrnn_supervisor.py +++ b/model/pytorch/dcrnn_supervisor.py @@ -154,7 +154,7 @@ class DCRNNSupervisor: torch.nn.utils.clip_grad_norm_(self.dcrnn_model.parameters(), self.max_grad_norm) optimizer.step() - self._logger.info("finished one batch in {}".format(time.time() - start_time)) + self._logger.info("finished one batch in {:.1f}s".format(time.time() - start_time)) lr_scheduler.step()