From b6a2b3fe8eeb3d9573a5cdf80dbac7ec772f8750 Mon Sep 17 00:00:00 2001 From: Chintan Shah Date: Sun, 6 Oct 2019 14:34:58 -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 84d0938..f593c50 100644 --- a/model/pytorch/dcrnn_supervisor.py +++ b/model/pytorch/dcrnn_supervisor.py @@ -129,6 +129,7 @@ class DCRNNSupervisor: self.dcrnn_model = self.dcrnn_model.train() self._logger.info('Start training ...') + self._logger.info("num_batches:".format(self._data['train_loader'].num_batch)) for epoch_num in range(epochs): train_iterator = self._data['train_loader'].get_iterator() losses = [] @@ -154,7 +155,6 @@ 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 {:.1f}s".format(time.time() - start_time)) lr_scheduler.step()