Added area_scaling config to cerra config

metno · Nov 8, 2024 · 9575cce · 9575cce
1 parent 9423f67
commit 9575cce
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/anemoi/training/config/stretched_grid_cerra.yaml b/src/anemoi/training/config/stretched_grid_cerra.yaml
@@ -141,3 +141,9 @@ training:
     rate: 5e-4  #8 * 0.625e-4
     min: 2.4e-6 #8 * 3e-7
     iterations: 150000
+
+  loss_scaling:
+    spatial:
+      _target_: anemoi.training.data.scaling.StretchedGridCutoutAreaWeights
+      target_nodes: ${graph.data}
+      cutout_weight_frac_of_global: 0.3