update model to work with PVNet=2.5 and datapipes=2.3

config.json

@@ -10,14 +10,16 @@
         0.98
     ],
     
-    "nwp_encoder": {
-      "_target_": "pvnet.models.multimodal.encoders.encoders3d.DefaultPVNet",
-      "_partial_": true,
-      "in_channels": 2,
-      "out_features": 256,
-      "number_of_conv3d_layers": 6,
-      "conv3d_channels": 32,
-      "image_size_pixels": 24
+    "nwp_encoders_dict": {
+        "ukv": {
+          "_target_": "pvnet.models.multimodal.encoders.encoders3d.DefaultPVNet",
+          "_partial_": true,
+          "in_channels": 2,
+          "out_features": 256,
+          "number_of_conv3d_layers": 6,
+          "conv3d_channels": 32,
+          "image_size_pixels": 24
+        }
     },
     
     "sat_encoder": {
@@ -47,8 +49,8 @@
     "forecast_minutes": 480,
     "history_minutes": 120,
     "sat_history_minutes": 90,
-    "nwp_history_minutes": 120,
-    "nwp_forecast_minutes": 480,
+    "nwp_history_minutes": {"ukv": 120},
+    "nwp_forecast_minutes": {"ukv": 480},
     "min_sat_delay_minutes": 60,
     
     "optimizer": {

data_config.yaml

@@ -9,7 +9,6 @@ input_data:
     forecast_minutes: 480
     gsp_zarr_path: PLACEHOLDER.zarr
     history_minutes: 120
-    metadata_only: false
     start_datetime: '1900-01-01T00:00:00'
     time_resolution_minutes: 30
   hrvsatellite:
@@ -22,15 +21,16 @@ input_data:
     hrvsatellite_zarr_path: ''
     time_resolution_minutes: 5
   nwp:
-    forecast_minutes: 480
-    history_minutes: 120
-    nwp_channels:
-    - t
-    - dswrf
-    nwp_image_size_pixels_height: 24
-    nwp_image_size_pixels_width: 24
-    nwp_zarr_path: PLACEHOLDER.zarr
-    time_resolution_minutes: 60
+    ukv:
+      forecast_minutes: 480
+      history_minutes: 120
+      nwp_channels:
+      - t
+      - dswrf
+      nwp_image_size_pixels_height: 24
+      nwp_image_size_pixels_width: 24
+      nwp_zarr_path: PLACEHOLDER.zarr
+      time_resolution_minutes: 60
   satellite:
     forecast_minutes: 0
     history_minutes: 90
@@ -51,5 +51,3 @@ input_data:
     satellite_image_size_pixels_width: 24
     satellite_zarr_path: PLACEHOLDER.zarr
     time_resolution_minutes: 5
-output_data:
-  filepath: not-needed

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:be7751ba901d23f35d4c25127d42783c00bc3ab206d5fba8723774afe0333d02
-size 46579073
+oid sha256:2701ff9c3e511fcab9932359d33411acf698b707a55de26e5e98c400eb5115fc
+size 46579902