Merge pull request #2240 from Zirunis/Zirunis-patch-1
Fix LR scheduler help in train.pypull/2243/head
commit
8b14fc7bb6
2
train.py
2
train.py
|
@ -210,7 +210,7 @@ group.add_argument('--opt-kwargs', nargs='*', default={}, action=utils.ParseKwar
|
|||
# Learning rate schedule parameters
|
||||
group = parser.add_argument_group('Learning rate schedule parameters')
|
||||
group.add_argument('--sched', type=str, default='cosine', metavar='SCHEDULER',
|
||||
help='LR scheduler (default: "step"')
|
||||
help='LR scheduler (default: "cosine"')
|
||||
group.add_argument('--sched-on-updates', action='store_true', default=False,
|
||||
help='Apply LR scheduler step on update instead of epoch end.')
|
||||
group.add_argument('--lr', type=float, default=None, metavar='LR',
|
||||
|
|
Loading…
Reference in New Issue