From a035bbdd0cc5e39954272d13d1e1595db738846a Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Wed, 10 Aug 2022 15:25:20 +0800
Subject: Fix a typo and add scheduler choices
---
simclr/evaluate.py | 3 ++-
simclr/main.py | 3 ++-
2 files changed, 4 insertions(+), 2 deletions(-)
(limited to 'simclr')
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)')
diff --git a/simclr/main.py b/simclr/main.py
index b91b42b..69e2ab2 100644
--- a/simclr/main.py
+++ b/simclr/main.py
@@ -93,8 +93,9 @@ def parse_args_and_config():
optim_group.add_argument('--weight-decay', default=1e-6, 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='warmup-anneal', type=str,
+ choices=('const', None, 'linear', 'warmup-anneal'),
help="Name of scheduler")
sched_group.add_argument('--warmup-iters', default=2344, type=int,
help='Epochs for warmup (`warmup-anneal` scheduler only)')
--
cgit v1.2.3