From a3a0bfb94e0755bc23b63e88c7bba2410afeacb5 Mon Sep 17 00:00:00 2001
From: Jacob Bieker <jacob@bieker.tech>
Date: Tue, 5 Dec 2023 10:24:01 +0000
Subject: [PATCH] Update configs for WindNet

---
 configs/callbacks/default.yaml     | 2 +-
 configs/config.yaml                | 2 +-
 configs/logger/wandb.yaml          | 4 ++--
 configs/model/wind_multimodal.yaml | 6 +++---
 configs/trainer/default.yaml       | 2 +-
 5 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/configs/callbacks/default.yaml b/configs/callbacks/default.yaml
index a832d219..c9483ab0 100644
--- 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
index e190caee..df388aa1 100644
--- 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
index bfd4d574..a724df0e 100644
--- 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
index ced5ea96..22d32810 100644
--- 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
index 5f051088..3f84fcd9 100644
--- 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