diff --git a/pytorch_lightning/utils/arg_parse.py b/pytorch_lightning/utils/arg_parse.py index d04f5253..0ccec472 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