From f246ae7fab5c4f1f56ddf07503f9b22f2ea8df0b Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 13 May 2019 21:55:57 -0400 Subject: [PATCH] added option to change default tensor --- pytorch_lightning/models/trainer.py | 1 - setup.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index d15b86f0..6cf38e49 100644 --- a/pytorch_lightning/models/trainer.py +++ b/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 diff --git a/setup.py b/setup.py index 59c20b26..f5370de4 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.dev1723', + version='0.1.dev1724', description="The Keras for ML researchers using PyTorch", author="William Falcon", author_email="waf2107@columbia.edu",