Skip to content

Commit 4ed93fc

Browse files
authored
Fix LR scheduler help in train.py
The default is and always has been the cosine scheduler, yet the help states that the default would be the step scheduler. Whatever the intended one was, for backwards compatibility the default should definitely remain cosine, which is why I changed the help comment to reflect that.
1 parent a1996ec commit 4ed93fc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@
210210
# Learning rate schedule parameters
211211
group = parser.add_argument_group('Learning rate schedule parameters')
212212
group.add_argument('--sched', type=str, default='cosine', metavar='SCHEDULER',
213-
help='LR scheduler (default: "step"')
213+
help='LR scheduler (default: "cosine"')
214214
group.add_argument('--sched-on-updates', action='store_true', default=False,
215215
help='Apply LR scheduler step on update instead of epoch end.')
216216
group.add_argument('--lr', type=float, default=None, metavar='LR',

0 commit comments

Comments
 (0)