diff --git a/pytorch_lightning/utilities/arg_parse.py b/pytorch_lightning/utilities/arg_parse.py index 91696433..f274e751 100644 --- a/pytorch_lightning/utilities/arg_parse.py +++ b/pytorch_lightning/utilities/arg_parse.py @@ -1,4 +1,7 @@ - +""" +List of default args which mught be useful for all the available flags +Might need to update with the new flags +""" def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None): @@ -73,4 +76,4 @@ def add_default_args(parser, root_dir, rand_seed=None, possible_model_names=None 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 + parser.add_argument('--lr_scheduler_milestones', default=None, type=str)