diff --git a/model/pytorch/dcrnn_supervisor.py b/model/pytorch/dcrnn_supervisor.py index f1692be..6a8462a 100644 --- 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: