From a8e57602d3a5539bd4a0e80567df6e01624f98bf Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 13 May 2019 22:03:47 -0400 Subject: [PATCH] added option to change default tensor --- pytorch_lightning/utils/arg_parse.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/utils/arg_parse.py b/pytorch_lightning/utils/arg_parse.py index 5cbeb4f126..fc971395a9 100644 --- a/pytorch_lightning/utils/arg_parse.py +++ b/pytorch_lightning/utils/arg_parse.py @@ -50,6 +50,7 @@ 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') # run on hpc parser.add_argument('--on_cluster', dest='on_cluster', action='store_true')