From 3f8e1333037f120cb1d759798afdcbb0f9f80435 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 18:13:01 -0400 Subject: [PATCH] updated args --- pytorch_lightning/utils/arg_parse.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/utils/arg_parse.py b/pytorch_lightning/utils/arg_parse.py index d04f525355..0ccec47299 100644 --- a/pytorch_lightning/utils/arg_parse.py +++ b/pytorch_lightning/utils/arg_parse.py @@ -69,8 +69,8 @@ def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None parser.add_argument('--random_seed', default=rand_seed, type=int) parser.add_argument('--live', dest='live', action='store_true', help='runs on gpu without cluster') - parser.add_argument('--enable_debug', dest='debug', action='store_true', help='enables/disables test tube') - parser.add_argument('--enable_local', dest='local', action='store_true', help='enables local tng') + parser.add_argument('--debug', dest='debug', action='store_true', help='enables/disables test tube') + parser.add_argument('--local', dest='local', action='store_true', help='enables local tng') # optimizer parser.add_argument('--lr_scheduler_milestones', default=None, type=str) \ No newline at end of file