diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 6cf38e49df..bbfa5b7316 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -32,6 +32,7 @@ class Trainer(TrainerIO): train_percent_check=1.0, val_percent_check=1.0, test_percent_check=1.0, val_check_interval=0.95, log_save_interval=1, add_log_row_interval=1, lr_scheduler_milestones=None, + use_amp=False, nb_sanity_val_steps=5): # Transfer params @@ -80,7 +81,6 @@ class Trainer(TrainerIO): print('gpu available: {}, used: {}'.format(torch.cuda.is_available(), self.on_gpu)) # apex test - use_amp = True self.use_amp = use_amp and APEX_AVAILABLE if self.use_amp: print('using 16bit precision') diff --git a/setup.py b/setup.py index f5370de4fd..74a2e0ae63 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.dev1724', + version='0.1.dev1725', description="The Keras for ML researchers using PyTorch", author="William Falcon", author_email="waf2107@columbia.edu",