diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-10 15:25:20 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-10 15:27:38 +0800 |
| commit | a035bbdd0cc5e39954272d13d1e1595db738846a (patch) | |
| tree | 4dd155b460232e26665a4a44e9f6e347f8e25ad7 /simclr/main.py | |
| parent | f9b51f152e94896a122325fa2c48b0ce10c881c5 (diff) | |
Fix a typo and add scheduler choices
Diffstat (limited to 'simclr/main.py')
| -rw-r--r-- | simclr/main.py | 3 |
1 files changed, 2 insertions, 1 deletions
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)') |