From cbb7cd4248c8f125323c532bc7c3337c606d2203 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Mon, 8 Aug 2022 19:17:20 +0800
Subject: Make pretraining checkpoint argument required

---
 simclr/evaluate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'simclr/evaluate.py')

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,
-- 
cgit v1.2.3