From a16171335b6d68d8578cae39ce54ab2e7d81b21d Mon Sep 17 00:00:00 2001 From: han Date: Tue, 10 May 2022 07:57:19 +0900 Subject: [PATCH] fix: change milestones to decay-milestones - change argparser option `milestone` to `decay-milestone` --- timm/scheduler/scheduler_factory.py | 2 +- train.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/timm/scheduler/scheduler_factory.py b/timm/scheduler/scheduler_factory.py index 2f5a49fa..3e100fe0 100644 --- a/timm/scheduler/scheduler_factory.py +++ b/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.milestones, + decay_t=args.decay_milestones, decay_rate=args.decay_rate, warmup_lr_init=args.warmup_lr, warmup_t=args.warmup_epochs, diff --git a/train.py b/train.py index ec8778d9..6f31e295 100755 --- a/train.py +++ b/train.py @@ -171,8 +171,8 @@ parser.add_argument('--epoch-repeats', type=float, default=0., metavar='N', 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-milestones', default=[30, 60], type=int, nargs='+', metavar="MILESTONES", + help='list of decay 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',