localization_metrics

EMCarrami · Nov 1, 2023 · 112a2f0 · 112a2f0
1 parent dfde895
commit 112a2f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/localization_metrics.py b/localization_metrics.py
@@ -94,7 +94,7 @@ def localization_metrics(
         default="What is the subcellular location of this protein?",
         help="Question prompt to use.",
     )
-    parser.add_argument("--checkpoint_path", type=int, help="path to model checkpoint")
+    parser.add_argument("--checkpoint_path", type=str, help="path to model checkpoint")
     args = parser.parse_args()
 
     localization_metrics(config_path=args.config_path, checkpoint_path=args.checkpoint_path, question=args.question)