From c3dc7ab4fbee602a153cc065c01a14aebc775445 Mon Sep 17 00:00:00 2001 From: Jacob Bieker Date: Tue, 5 Dec 2023 12:17:53 +0000 Subject: [PATCH] Debugging --- pvnet/models/multimodal/site_encoders/encoders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pvnet/models/multimodal/site_encoders/encoders.py b/pvnet/models/multimodal/site_encoders/encoders.py index 86793a87..93757ec4 100644 --- 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):