Debugging

openclimatefix · Dec 5, 2023 · 3c7b7aa · 3c7b7aa
1 parent ec12ee5
commit 3c7b7aa
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
@@ -329,7 +329,7 @@ def _encode_query(self, x):
         # Select the first one
         gsp_ids = x[BatchKey.sensor_id][:,0,:].squeeze().int()
         print(f"{gsp_ids.shape=}")
-        query = self.sensor_id_embedding(gsp_ids)  # .unsqueeze(1)
+        query = self.sensor_id_embedding(gsp_ids).unsqueeze(1)
         print(f"{query.shape=}")
         return query