added amp level option

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()
 

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",