diff --git a/pvnet/models/multimodal/site_encoders/encoders.py b/pvnet/models/multimodal/site_encoders/encoders.py
index 7fb2e7fb..f45411ed 100644
--- 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):