Upload folder using huggingface_hub

README.md

@@ -39,7 +39,7 @@ Data is prepared with the `ocf_datapipes.training.pvnet` datapipe [2].
 ## Results
 
 The training logs for the current model can be found here:
- - [https://wandb.ai/openclimatefix/pvnet2.1/runs/5llq8iw6](https://wandb.ai/openclimatefix/pvnet2.1/runs/5llq8iw6)
+ - [https://wandb.ai/openclimatefix/pvnet2.1/runs/xdlew7ib](https://wandb.ai/openclimatefix/pvnet2.1/runs/xdlew7ib)
 
 
 The training logs for all model runs of PVNet2 can be found [here](https://wandb.ai/openclimatefix/pvnet2.1).

config.json

@@ -15,31 +15,31 @@
         "ecmwf": {
             "_target_": "pvnet.models.multimodal.encoders.encoders3d.ResConv3DNet2",
             "_partial_": true,
-            "in_channels": 6,
+            "in_channels": 7,
             "out_features": 256,
             "n_res_blocks": 4,
             "hidden_channels": 256,
-            "image_size_pixels": 64
+            "image_size_pixels": 84
         }
     },
     "wind_encoder": {
         "_target_": "pvnet.models.multimodal.site_encoders.encoders.SingleAttentionNetwork",
         "_partial_": true,
         "num_sites": 1,
-        "out_features": 40,
+        "out_features": 64,
         "num_heads": 4,
-        "kdim": 40,
-        "id_embed_dim": 20
+        "kdim": 64,
+        "id_embed_dim": 64
     },
     "output_network": {
         "_target_": "pvnet.models.multimodal.linear_networks.networks.ResFCNet2",
         "_partial_": true,
-        "fc_hidden_features": 256,
+        "fc_hidden_features": 1024,
         "n_res_blocks": 6,
-        "res_block_layers": 2,
+        "res_block_layers": 4,
         "dropout_frac": 0.0
     },
-    "embedding_dim": 32,
+    "embedding_dim": 128,
     "include_sun": false,
     "include_gsp_yield_history": false,
     "forecast_minutes": 2880,
@@ -53,6 +53,7 @@
         "ecmwf": 2880
     },
     "wind_history_minutes": 60,
+    "sensor_interval_minutes": 15,
     "target_key": "wind",
     "optimizer": {
         "_target_": "pvnet.optimizers.EmbAdamWReduceLROnPlateau",

data_config.yaml

@@ -7,22 +7,47 @@ input_data:
   default_history_minutes: 60
   nwp:
     ecmwf:
+      dropout_fraction: 1.0
+      dropout_timedeltas_minutes:
+      - -360
       forecast_minutes: 2880
       history_minutes: 60
       nwp_channels:
+      - t2m
       - u10
       - u100
       - u200
       - v10
       - v100
       - v200
-      nwp_image_size_pixels_height: 64
-      nwp_image_size_pixels_width: 64
+      nwp_image_size_pixels_height: 168
+      nwp_image_size_pixels_width: 168
       nwp_provider: ecmwf
       nwp_zarr_path: PLACEHOLDER.zarr
       time_resolution_minutes: 60
       x_dim_name: longitude
       y_dim_name: latitude
+  sensor:
+    forecast_minutes: 2880
+    get_center: false
+    history_minutes: 60
+    sensor_filename: /mnt/storage_b/nwp/meteomatics/nw_india/wind*.zarr.zip
+    sensor_variables:
+    - 100u
+    - 100v
+    - 10u
+    - 10v
+    - 200u
+    - 200v
+    - cape:Jkg
+    - air_density_25m:kgm3
+    - air_density_10m:kgm3
+    - air_density_100m:kgm3
+    - air_density_200m:kgm3
+    - wind_gusts_200m:ms
+    - wind_gusts_100m:ms
+    - wind_gusts_10m:ms
+    time_resolution_minutes: 15
   wind:
     get_center: true
     n_wind_systems_per_example: 1

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2b191f52cdc8932d02ce45718bb80cd719b7ba98a9d81214316a936f164bedeb
-size 271691706
+oid sha256:f4857cdf19f63c6b49a59d3a8bf3caecb21b2341cc5ef5091e8b0c85817d6d63
+size 493586506