wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update arg_parse.py

Browse Source
This commit is contained in:
William Falcon
2019-08-05 14:14:28 -04:00
committed by GitHub
parent 2b31ced060
commit 01caefb4e6
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/utilities/arg_parse.py
+5 -2
View File
@@ -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)
parser.add_argument('--lr_scheduler_milestones', default=None, type=str)