Merge pull request #2240 from Zirunis/Zirunis-patch-1

Fix LR scheduler help in train.py
pull/2243/head
Ross Wightman 2024-07-23 11:04:18 -07:00 committed by GitHub
commit 8b14fc7bb6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -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',