From eb13bb8313e613a4b6d76ecc62e99d9c17eef383 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 May 2019 15:58:58 -0400 Subject: [PATCH] added amp level option --- pytorch_lightning/utils/arg_parse.py | 4 ++-- setup.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/utils/arg_parse.py b/pytorch_lightning/utils/arg_parse.py index 4d05e297cb..d04f525355 100644 --- a/pytorch_lightning/utils/arg_parse.py +++ b/pytorch_lightning/utils/arg_parse.py @@ -50,8 +50,8 @@ def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None parser.add_argument('--single_run_gpu', dest='single_run_gpu', action='store_true') parser.add_argument('--disable_cuda', dest='disable_cuda', action='store_true') parser.add_argument('--default_tensor_type', default='torch.cuda.FloatTensor', type=str) - parser.add_argument('--use_amp', dest='use_amp', action='store_true')a= - parser.add_argument('--check_grad_nans', dest='check_grad_nans', action='store_true')a= + parser.add_argument('--use_amp', dest='use_amp', action='store_true') + parser.add_argument('--check_grad_nans', dest='check_grad_nans', action='store_true') parser.add_argument('--amp_level', default='O2',type=str) diff --git a/setup.py b/setup.py index 8f49361c57..18f211c101 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.dev1829', + version='0.1.dev1830', description="The Keras for ML researchers using PyTorch", author="William Falcon", author_email="waf2107@columbia.edu",