added option to change default tensor

pytorch_lightning/models/trainer.py

@@ -223,7 +223,6 @@ class Trainer(TrainerIO):
             # An example
             self.model, optimizer = amp.initialize(
                 self.model, self.optimizers[0], opt_level="O2",
-                keep_batchnorm_fp32=True, loss_scale="dynamic"
             )
             self.optimizers[0] = optimizer
             model.trainer = self

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.dev1723',
+    version='0.1.dev1724',
     description="The Keras for ML researchers using PyTorch",
     author="William Falcon",
     author_email="waf2107@columbia.edu",