Merge pull request #716 from ACEsuit/develop

add option pt_train_file == "mp"
ACEsuit · Nov 25, 2024 · cf6a4b5 · cf6a4b5
2 parents 2587430 + ec43304
commit cf6a4b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mace/cli/run_train.py b/mace/cli/run_train.py
@@ -265,7 +265,7 @@ def run(args: argparse.Namespace) -> None:
         args.loss = "universal"
         if (
             args.foundation_model in ["small", "medium", "large"]
-            or args.pt_train_file is None
+            or args.pt_train_file == "mp"
         ):
             logging.info(
                 "Using foundation model for multiheads finetuning with Materials Project data"