Merge pull request #689 from ACEsuit/develop

fix name in foundation head args
ACEsuit · Nov 12, 2024 · 8ffece8 · 8ffece8
2 parents 624e557 + c57b6da
commit 8ffece8
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,