From a035bbdd0cc5e39954272d13d1e1595db738846a Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Wed, 10 Aug 2022 15:25:20 +0800 Subject: Fix a typo and add scheduler choices --- simclr/evaluate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'simclr/evaluate.py') diff --git a/simclr/evaluate.py b/simclr/evaluate.py index 3267ead..f4a8fda 100644 --- a/simclr/evaluate.py +++ b/simclr/evaluate.py @@ -84,8 +84,9 @@ def parse_args_and_config(): optim_group.add_argument('--weight-decay', default=0., type=float, help='Weight decay (l2 regularization)') - sched_group = parser.add_argument_group('Optimizer parameters') + sched_group = parser.add_argument_group('Scheduler parameters') sched_group.add_argument('--sched', default=None, type=str, + choices=('const', None, 'linear', 'warmup-anneal'), help="Name of scheduler") sched_group.add_argument('--warmup-iters', default=5, type=int, help='Epochs for warmup (`warmup-anneal` scheduler only)') -- cgit v1.2.3