fix disable knn default value (#142)

vturrisi · Sep 21, 2021 · e9be48a · e9be48a
1 parent af7282f
commit e9be48a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/solo/methods/base.py b/solo/methods/base.py
@@ -293,7 +293,7 @@ def add_model_specific_args(parent_parser: ArgumentParser) -> ArgumentParser:
         parser.add_argument("--encode_indexes_into_labels", action="store_true")
 
         # online knn eval
-        parser.add_argument("--disable_knn_eval", default=True, action="store_false")
+        parser.add_argument("--disable_knn_eval", action="store_true")
         parser.add_argument("--knn_k", default=20, type=int)
 
         return parent_parser