diff --git a/model/pytorch/dcrnn_supervisor.py b/model/pytorch/dcrnn_supervisor.py index 0b36519..2355990 100644 --- a/model/pytorch/dcrnn_supervisor.py +++ b/model/pytorch/dcrnn_supervisor.py @@ -197,8 +197,6 @@ class DCRNNSupervisor: val_loss = self.evaluate(dataset='val', batches_seen=batches_seen) - self.dcrnn_model = self.dcrnn_model.train() - end_time = time.time() self._writer.add_scalar('training loss',