From 2ab5831c74141c682453e528327449709908b05d Mon Sep 17 00:00:00 2001
From: Luis Barroso-Luque <lbluque@users.noreply.github.com>
Date: Fri, 7 Feb 2025 11:33:13 -0800
Subject: [PATCH] update reduction to mean (#1005)

---
 configs/omat24/all/eqV2_153M.yml                    | 2 +-
 configs/omat24/all/eqV2_31M.yml                     | 2 +-
 configs/omat24/all/eqV2_86M.yml                     | 2 +-
 configs/omat24/finetune/eqV2_153M_ft_salexmptrj.yml | 2 +-
 configs/omat24/finetune/eqV2_31M_ft_salexmptrj.yml  | 2 +-
 configs/omat24/finetune/eqV2_86M_ft_salexmptrj.yml  | 2 +-
 configs/omat24/mptrj/eqV2_153M_dens_mptrj.yml       | 2 +-
 configs/omat24/mptrj/eqV2_31M_dens_mptrj.yml        | 2 +-
 configs/omat24/mptrj/eqV2_31M_mptrj.yml             | 2 +-
 configs/omat24/mptrj/eqV2_86M_dens_mptrj.yml        | 2 +-
 10 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/configs/omat24/all/eqV2_153M.yml b/configs/omat24/all/eqV2_153M.yml
index bf87a9e3ea..c457a34441 100644
--- a/configs/omat24/all/eqV2_153M.yml
+++ b/configs/omat24/all/eqV2_153M.yml
@@ -52,7 +52,7 @@ loss_functions:
       coefficient: 5
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 5
 
 optim:
diff --git a/configs/omat24/all/eqV2_31M.yml b/configs/omat24/all/eqV2_31M.yml
index 902a58b31c..50003ee1e6 100644
--- a/configs/omat24/all/eqV2_31M.yml
+++ b/configs/omat24/all/eqV2_31M.yml
@@ -53,7 +53,7 @@ loss_functions:
       coefficient: 5
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 5
 
 
diff --git a/configs/omat24/all/eqV2_86M.yml b/configs/omat24/all/eqV2_86M.yml
index 154ed57cb4..cb3a9a2e5e 100644
--- a/configs/omat24/all/eqV2_86M.yml
+++ b/configs/omat24/all/eqV2_86M.yml
@@ -52,7 +52,7 @@ loss_functions:
       coefficient: 5
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 5
 
 optim:
diff --git a/configs/omat24/finetune/eqV2_153M_ft_salexmptrj.yml b/configs/omat24/finetune/eqV2_153M_ft_salexmptrj.yml
index bd04e683d8..c43d73463b 100644
--- a/configs/omat24/finetune/eqV2_153M_ft_salexmptrj.yml
+++ b/configs/omat24/finetune/eqV2_153M_ft_salexmptrj.yml
@@ -54,7 +54,7 @@ loss_functions:
       coefficient: 1
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 1
 
 model:
diff --git a/configs/omat24/finetune/eqV2_31M_ft_salexmptrj.yml b/configs/omat24/finetune/eqV2_31M_ft_salexmptrj.yml
index 36e89c66bd..33430bd320 100644
--- a/configs/omat24/finetune/eqV2_31M_ft_salexmptrj.yml
+++ b/configs/omat24/finetune/eqV2_31M_ft_salexmptrj.yml
@@ -54,7 +54,7 @@ loss_functions:
       coefficient: 1
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 1
 
 model:
diff --git a/configs/omat24/finetune/eqV2_86M_ft_salexmptrj.yml b/configs/omat24/finetune/eqV2_86M_ft_salexmptrj.yml
index 8e230aa1a6..0d04a7c085 100644
--- a/configs/omat24/finetune/eqV2_86M_ft_salexmptrj.yml
+++ b/configs/omat24/finetune/eqV2_86M_ft_salexmptrj.yml
@@ -52,7 +52,7 @@ loss_functions:
       coefficient: 1
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 1
 
 hide_eval_progressbar: False
diff --git a/configs/omat24/mptrj/eqV2_153M_dens_mptrj.yml b/configs/omat24/mptrj/eqV2_153M_dens_mptrj.yml
index 435309a224..646c42cd56 100644
--- a/configs/omat24/mptrj/eqV2_153M_dens_mptrj.yml
+++ b/configs/omat24/mptrj/eqV2_153M_dens_mptrj.yml
@@ -54,7 +54,7 @@ loss_functions:
       coefficient: 5
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 5
 
 optim:
diff --git a/configs/omat24/mptrj/eqV2_31M_dens_mptrj.yml b/configs/omat24/mptrj/eqV2_31M_dens_mptrj.yml
index 78ccb4b92e..f8c66da3f3 100644
--- a/configs/omat24/mptrj/eqV2_31M_dens_mptrj.yml
+++ b/configs/omat24/mptrj/eqV2_31M_dens_mptrj.yml
@@ -54,7 +54,7 @@ loss_functions:
       coefficient: 5
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 5
 
 optim:
diff --git a/configs/omat24/mptrj/eqV2_31M_mptrj.yml b/configs/omat24/mptrj/eqV2_31M_mptrj.yml
index 7f4c83cf68..57a0108c46 100644
--- a/configs/omat24/mptrj/eqV2_31M_mptrj.yml
+++ b/configs/omat24/mptrj/eqV2_31M_mptrj.yml
@@ -54,7 +54,7 @@ loss_functions:
       coefficient: 5
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 5
 
 optim:
diff --git a/configs/omat24/mptrj/eqV2_86M_dens_mptrj.yml b/configs/omat24/mptrj/eqV2_86M_dens_mptrj.yml
index 47f0958856..bb735993ef 100644
--- a/configs/omat24/mptrj/eqV2_86M_dens_mptrj.yml
+++ b/configs/omat24/mptrj/eqV2_86M_dens_mptrj.yml
@@ -54,7 +54,7 @@ loss_functions:
       coefficient: 5
   - stress_anisotropic:
       fn: mae
-      reduction: mean_all
+      reduction: mean
       coefficient: 5
 
 optim: