From b59af1813b71ad5c9581bddc83d89b35cd0de3b7 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Tue, 25 Jun 2019 19:56:12 -0400
Subject: [PATCH] updated args

---
 docs/source/examples/example_model.py | 3 ---
 pytorch_lightning/models/trainer.py   | 1 +
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py
index 9f48e6f363..6570c5d31e 100644
--- a/docs/source/examples/example_model.py
+++ b/docs/source/examples/example_model.py
@@ -42,7 +42,6 @@ class ExampleModel(RootModule):
     # ---------------------
     def forward(self, x, a):
 
-        pdb.set_trace()
         x = self.c_d1(x)
         x = F.tanh(x)
         x = self.c_d1_bn(x)
@@ -63,7 +62,6 @@ class ExampleModel(RootModule):
         :param data_batch:
         :return:
         """
-        pdb.set_trace()
         # forward pass
         x, y = data_batch
         x = x.view(x.size(0), -1)
@@ -81,7 +79,6 @@ class ExampleModel(RootModule):
         :param data_batch:
         :return:
         """
-        pdb.set_trace()
         x, y = data_batch
         x = x.view(x.size(0), -1)
         y_hat = self.forward(x)
diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index a9c7462b59..1476064fe1 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -188,6 +188,7 @@ class Trainer(TrainerIO):
             # RUN VALIDATION STEP
             # -----------------
             output = model(data_batch, batch_i)
+            pdb.set_trace()
             outputs.append(output)
 
             # batch done