update config for new features

NCAR · Oct 17, 2024 · ee84246 · ee84246
1 parent a7a12a5
commit ee84246
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/config/example_physics_single.yml b/config/example_physics_single.yml
@@ -151,13 +151,19 @@ model:
     window_size: 7          # window size (default: 7)
     depth: 2               # number of swin transformers (default: 48)
 
-    # map boundary padding
-    pad_lon: 40             # number of grids to pad on 0 and 360 deg lon
-    pad_lat: 40             # number of grids to pad on -90 and 90 deg lat
-
     # use spectral norm
     use_spectral_norm: True
 
+    # use interpolation to match the output size
+    interp: True
+
+    # map boundary padding
+    padding_conf:
+        activate: True
+        mode: mirror
+        pad_lat: 40
+        pad_lon: 40
+
     post_conf: 
         activate: True
 

diff --git a/config/wxformer_1dg_test.yml b/config/wxformer_1dg_test.yml
@@ -125,9 +125,15 @@ model:
     attn_dropout: 0.                  # Dropout probability for attention layers (default: 0.0)
     ff_dropout: 0.                    # Dropout probability for feed-forward layers (default: 0.0)
 
+    # use interpolation to match the output size
+    interp: True
+
     # map boundary padding
-    pad_lon: 48             # number of grids to pad on 0 and 360 deg lon
-    pad_lat: 48             # number of grids to pad on -90 and 90 deg lat
+    padding_conf:
+        activate: True
+        mode: earth
+        pad_lat: 48
+        pad_lon: 48
 
 loss: 
     # the main training loss