From 4ed93fce9345a268e0d6b0df92aa687e71a26d4e Mon Sep 17 00:00:00 2001
From: Zirunis <122003158+Zirunis@users.noreply.github.com>
Date: Mon, 22 Jul 2024 23:04:00 +0200
Subject: [PATCH] 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.
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 2110f98d..09968494 100755
--- a/train.py
+++ b/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',