diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 9744a9d3..6a83356f 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -378,9 +378,9 @@ class Trainer(TrainerIO):
         else:
             loss.backward()
 
-            if self.check_grad_nans:
-                for param in self.model.parameters():
-                    print(param.grad.float().sum())
+        if self.check_grad_nans:
+            for param in self.model.parameters():
+                print(param.grad.float().sum())
 
         self.batch_loss_value += loss.item()
 
diff --git a/setup.py b/setup.py
index 18f211c1..137e37cf 100755
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@ from setuptools import setup, find_packages
 # http://blog.ionelmc.ro/2014/05/25/python-packaging/
 setup(
     name="pytorch-lightning",
-    version='0.1.dev1830',
+    version='0.1.dev1831',
     description="The Keras for ML researchers using PyTorch",
     author="William Falcon",
     author_email="waf2107@columbia.edu",