default for get task metrics

FAIR-Chem · Jul 18, 2023 · c9f3980 · c9f3980
1 parent 9d0fc41
commit c9f3980
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ocpmodels/trainers/base_trainer.py b/ocpmodels/trainers/base_trainer.py
@@ -179,7 +179,7 @@ def __init__(
         self.evaluator = Evaluator(
             task=name,
             eval_metrics=self.config["task"].get(
-                "evaluation_metrics", Evaluator.task_metrics[name]
+                "evaluation_metrics", Evaluator.task_metrics.get(name, {})
             ),
         )
 
@@ -952,7 +952,7 @@ def validate(self, split="val", disable_tqdm=False):
         evaluator = Evaluator(
             task=self.name,
             eval_metrics=self.config["task"].get(
-                "evaluation_metrics", Evaluator.task_metrics[self.name]
+                "evaluation_metrics", Evaluator.task_metrics.get(self.name, {})
             ),
         )