fix: update configs

ecmwf · Sep 24, 2024 · 4705130 · 4705130
1 parent 9323561
commit 4705130
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/anemoi/training/config/graph/multiencoder_lam_multi_scale.yaml b/src/anemoi/training/config/graph/multiencoder_lam_multi_scale.yaml
@@ -9,13 +9,12 @@ nodes:
   data:
     node_builder:
       _target_: anemoi.graphs.nodes.ZarrDatasetNodes # options: ZarrDatasetNodes, NPZFileNodes
-      dataset: ${dataloader.dataset}
+      dataset: ${hardware.paths.data}/${hardware.files.dataset}
     attributes: ${graph.attributes.nodes}
   boundary:
     node_builder:
       _target_: anemoi.graphs.nodes.ZarrDatasetNodes
       dataset: ${hardware.paths.data}/${hardware.files.forcing_dataset}
-      reference_node_name: ${graph.data}
     attributes: ${graph.attributes.nodes}
   # Hidden nodes
   hidden:
@@ -48,7 +47,6 @@ edges:
 # Decoder configuration
 - source_name: ${graph.hidden}
   target_name: ${graph.data}
-  target_mask_attr_name: cutout
   edge_builder:
     _target_: anemoi.graphs.edges.KNNEdges # options: KNNEdges, CutOffEdges
     num_nearest_neighbours: 3 # only for knn method
@@ -58,7 +56,7 @@ edges:
 attributes:
   nodes:
     area_weight:
-      _target_: anemoi.graphs.nodes.attributes.AreaWeights # options: Area, Uniform
+      _target_: anemoi.graphs.nodes.attributes.UniformWeights # options: Area, Uniform
       #norm: unit-max # options: l1, l2, unit-max, unit-sum, unit-std
   edges:
     edge_length:

diff --git a/src/anemoi/training/data/datamodule.py b/src/anemoi/training/data/datamodule.py
@@ -171,7 +171,7 @@ def _get_dataset(
             shuffle=shuffle,
             label=label,
         )
-        self._check_resolution(data.resolution)
+        #self._check_resolution(data.resolution)
         return data
 
     def _get_dataloader(self, ds: NativeGridDataset, stage: str) -> DataLoader: