Debugging

openclimatefix · Dec 5, 2023 · 65f3e6a · 65f3e6a
1 parent d2c52f7
commit 65f3e6a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pvnet/models/multimodal/site_encoders/encoders.py b/pvnet/models/multimodal/site_encoders/encoders.py
@@ -327,7 +327,9 @@ def __init__(
 
     def _encode_query(self, x):
         gsp_ids = x[BatchKey.sensor_id].squeeze().int()
+        print(f"{gsp_ids.shape=}")
         query = self.sensor_id_embedding(gsp_ids)  # .unsqueeze(1)
+        print(f"{query.shape=}")
         return query
 
     def _encode_key(self, x):