Update config

openclimatefix · Nov 27, 2023 · 03d5b78 · 03d5b78
1 parent 1713266
commit 03d5b78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/configs/model/multimodal.yaml b/configs/model/multimodal.yaml
@@ -48,13 +48,13 @@ pv_encoder:
 #--------------------------------------------
 
 sensor_encoder:
-  _target_: pvnet.models.multimodal.site_encoders.encoders.SingleAttentionNetwork
+  _target_: pvnet.models.multimodal.site_encoders.encoders.SingleSensorAttentionNetwork
   _partial_: True
   num_sites: 123
   out_features: 40
   num_heads: 4
   kdim: 40
-  pv_id_embed_dim: 20
+  sensor_id_embed_dim: 20
 
 #--------------------------------------------
 # Tabular network settings