Debugging

openclimatefix · Dec 5, 2023 · c3dc7ab · c3dc7ab
1 parent 4f944f1
commit c3dc7ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet/models/multimodal/site_encoders/encoders.py b/pvnet/models/multimodal/site_encoders/encoders.py
@@ -327,7 +327,7 @@ def __init__(
 
     def _encode_query(self, x):
         gsp_ids = x[BatchKey.sensor_id].squeeze().int()
-        query = self.sensor_id_embedding(gsp_ids).unsqueeze(1)
+        query = self.sensor_id_embedding(gsp_ids) #.unsqueeze(1)
         return query
 
     def _encode_key(self, x):