fix name in foundation head args

ilyes319 · ilyes319 · commit c57b6dac8922 · 2024-11-12T20:12:51.000Z
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,

Original file line number	Diff line number	Diff line change
`@@ -361,7 +361,7 @@ def build_default_arg_parser() -> argparse.ArgumentParser:`
`361`	`361`	`default=True,`
`362`	`362`	`)`
`363`	`363`	`parser.add_argument(`
`364`		`- "--foundation_model_head",`
	`364`	`+ "--foundation_head",`
`365`	`365`	`help="Name of the head to use for fine-tuning",`
`366`	`366`	`type=str,`
`367`	`367`	`default=None,`