From 078bbc5df510da44182b15861f74bf1862d1a4a5 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 20:19:11 -0400 Subject: [PATCH] updated args --- pytorch_lightning/models/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index c1152211e6..671512f834 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -209,8 +209,6 @@ class Trainer(TrainerIO): if self.enable_tqdm and self.prog_bar is not None: self.prog_bar.update(1) - pdb.set_trace() - # give model a chance to do something with the outputs val_results = model.module.validation_end(outputs) @@ -219,6 +217,8 @@ class Trainer(TrainerIO): # enable gradients to save memory torch.set_grad_enabled(True) + + pdb.set_trace() return val_results def __get_dataloaders(self, model):