default for get task metrics

FAIR-Chem · Jul 18, 2023 · 99eb482 · 99eb482
1 parent 4961bb1
commit 99eb482
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ocpmodels/trainers/base_trainer.py b/ocpmodels/trainers/base_trainer.py
@@ -181,7 +181,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, {})
             ),
         )
 
@@ -960,7 +960,7 @@ def validate(self, split: str = "val", disable_tqdm: bool = 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, {})
             ),
         )