Update configs for WindNet

openclimatefix · Dec 5, 2023 · a3a0bfb · a3a0bfb
1 parent 3370353
commit a3a0bfb
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 8 deletions.
diff --git a/configs/callbacks/default.yaml b/configs/callbacks/default.yaml
@@ -34,7 +34,7 @@ model_checkpoint:
   every_n_epochs: 1
   verbose: False
   filename: "epoch={epoch}-step={step}"
-  dirpath: "checkpoints/pvnet2.1/${model_name}" #${..model_name}
+  dirpath: "checkpoints/windnet2.1/${model_name}" #${..model_name}
   auto_insert_metric_name: False
   save_on_train_epoch_end: False
 #device_stats_monitor:

diff --git a/configs/config.yaml b/configs/config.yaml
@@ -3,7 +3,7 @@
 # specify here default training configuration
 defaults:
   - trainer: default.yaml
-  - model: multimodal.yaml
+  - model: wind_multimodal.yaml
   - datamodule: ocf_datapipes.yaml
   - callbacks: default.yaml # set this to null if you don't want to use callbacks
   - logger: wandb.yaml # set logger here or use command line (e.g. `python run.py logger=wandb`)

diff --git a/configs/logger/wandb.yaml b/configs/logger/wandb.yaml
@@ -2,9 +2,9 @@
 
 wandb:
   _target_: lightning.pytorch.loggers.wandb.WandbLogger
-  project: "pvnet2.1"
+  project: "windnet2.1"
   name: "${model_name}"
-  save_dir: "/mnt/disks/batches/"
+  save_dir: "/mnt/storage_ssd_4tb/windnet_batches/"
   offline: False # set True to store all logs only locally
   id: null # pass correct id to resume experiment!
   # entity: ""  # set to name of your wandb team or just remove it

diff --git a/configs/model/wind_multimodal.yaml b/configs/model/wind_multimodal.yaml
@@ -9,11 +9,11 @@ output_quantiles: [0.02, 0.1, 0.25, 0.5, 0.75, 0.9, 0.98]
 nwp_encoder:
   _target_: pvnet.models.multimodal.encoders.encoders3d.DefaultPVNet
   _partial_: True
-  in_channels: 2
+  in_channels: 14
   out_features: 256
   number_of_conv3d_layers: 6
   conv3d_channels: 32
-  image_size_pixels: 24
+  image_size_pixels: 16
 
 #--------------------------------------------
 # Sensor encoder settings
@@ -22,7 +22,7 @@ nwp_encoder:
 sensor_encoder:
   _target_: pvnet.models.multimodal.site_encoders.encoders.SingleSensorAttentionNetwork
   _partial_: True
-  num_sites: 123
+  num_sites: 12
   out_features: 40
   num_heads: 4
   kdim: 40

diff --git a/configs/trainer/default.yaml b/configs/trainer/default.yaml
@@ -1,7 +1,7 @@
 _target_: lightning.pytorch.trainer.trainer.Trainer
 
 # set `1` to train on GPU, `0` to train on CPU only
-accelerator: gpu
+accelerator: cpu
 devices: auto
 
 min_epochs: null