diff options
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, |