From b9d53971961c2f7751436a714b92a491a43c6806 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 18:14:48 -0400 Subject: [PATCH] updated args --- pytorch_lightning/utils/arg_parse.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pytorch_lightning/utils/arg_parse.py b/pytorch_lightning/utils/arg_parse.py index 0ccec472..d3018636 100644 --- a/pytorch_lightning/utils/arg_parse.py +++ b/pytorch_lightning/utils/arg_parse.py @@ -45,10 +45,8 @@ def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None parser.add_argument('--log_stdout', dest='log_stdout', action='store_true') # GPU - parser.add_argument('--per_experiment_nb_gpus', default=1, type=int) - parser.add_argument('--gpus', default='0', type=str) + parser.add_argument('--gpus', default=None, type=str) 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') parser.add_argument('--check_grad_nans', dest='check_grad_nans', action='store_true')