logging and refactor

2019-10-06 14:40:53 -04:00 · 2019-10-06 14:40:53 -04:00 · 55a087ac9f
parent 02c4681ad9
commit 55a087ac9f
1 changed files with 2 additions and 2 deletions
--- a/model/pytorch/dcrnn_supervisor.py
+++ b/model/pytorch/dcrnn_supervisor.py
@ -155,9 +155,9 @@ class DCRNNSupervisor:
                torch.nn.utils.clip_grad_norm_(self.dcrnn_model.parameters(), self.max_grad_norm)

                optimizer.step()
-
+            self._logger.info("epoch complete")
            lr_scheduler.step()
-
+            self._logger.info("evaluating now!")
            val_loss = self.evaluate(dataset='val')
            end_time = time.time()
            if epoch_num % log_every == 0: