diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-08 19:17:20 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-08 19:17:20 +0800 |
| commit | cbb7cd4248c8f125323c532bc7c3337c606d2203 (patch) | |
| tree | 6f55eda6f6616e9f98e786af6d4074269783fb1c /simclr | |
| parent | 86ea2427e845d09814dde9a68d732d3b1001f9fe (diff) | |
Make pretraining checkpoint argument required
Diffstat (limited to 'simclr')
| -rw-r--r-- | simclr/evaluate.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/simclr/evaluate.py b/simclr/evaluate.py index c029ffe..3267ead 100644 --- a/simclr/evaluate.py +++ b/simclr/evaluate.py @@ -44,7 +44,7 @@ def parse_args_and_config(): help='Number of dimension of embedding') parser.add_argument('--out-dim', default=128, type=int, help='Number of dimension after projection') - parser.add_argument('--pretrained-checkpoint', type=str, + parser.add_argument('--pretrained-checkpoint', type=str, required=True, help='Pretrained checkpoint location') parser.add_argument('--finetune', default=False, action=argparse.BooleanOptionalAction, |