fix name in foundation head args

ACEsuit · Nov 12, 2024 · c57b6da · c57b6da
1 parent b32b3e2
commit c57b6da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mace/tools/arg_parser.py b/mace/tools/arg_parser.py
@@ -361,7 +361,7 @@ def build_default_arg_parser() -> argparse.ArgumentParser:
         default=True,
     )
     parser.add_argument(
-        "--foundation_model_head",
+        "--foundation_head",
         help="Name of the head to use for fine-tuning",
         type=str,
         default=None,