fix: multistep lr decay epoch bugs

- add milestones arguments
- change decay_epochs to milestones variable

Author: hankyul2

timm/scheduler/scheduler_factory.py

@@ -71,7 +71,7 @@ def create_scheduler(args, optimizer):
     elif args.sched == 'multistep':
         lr_scheduler = MultiStepLRScheduler(
             optimizer,
-            decay_t=args.decay_epochs,
+            decay_t=args.milestones,
             decay_rate=args.decay_rate,
             warmup_lr_init=args.warmup_lr,
             warmup_t=args.warmup_epochs,

train.py

@@ -171,6 +171,8 @@
                     help='epoch repeat multiplier (number of times to repeat dataset epoch per train epoch).')
 parser.add_argument('--start-epoch', default=None, type=int, metavar='N',
                     help='manual epoch number (useful on restarts)')
+parser.add_argument('--milestones', default=[30, 60], type=int, nargs='+', metavar="MILESTONES",
+                    help='list of epoch indices for multistep lr. must be increasing')
 parser.add_argument('--decay-epochs', type=float, default=100, metavar='N',
                     help='epoch interval to decay LR')
 parser.add_argument('--warmup-epochs', type=int, default=3, metavar='N',