From 02c4681ad9742be1eaad12d9a18f18e245473051 Mon Sep 17 00:00:00 2001 From: Chintan Shah Date: Sun, 6 Oct 2019 14:38:44 -0400 Subject: [PATCH] logging and refactor --- model/pytorch/dcrnn_supervisor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model/pytorch/dcrnn_supervisor.py b/model/pytorch/dcrnn_supervisor.py index f593c50..f1692be 100644 --- a/model/pytorch/dcrnn_supervisor.py +++ b/model/pytorch/dcrnn_supervisor.py @@ -129,7 +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)) + 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 = [] @@ -144,7 +144,7 @@ class DCRNNSupervisor: output = self.dcrnn_model(x, y, batches_seen) loss = self._compute_loss(y, output, criterion) - self._logger.info(loss.item()) + self._logger.debug(loss.item()) losses.append(loss.item())