add: note

2026-05-13 07:39:53 +00:00 · 2025-09-27 12:13:53 +02:00
1 changed files with 3 additions and 1 deletions
@@ -59,7 +59,9 @@ class SpatialLearnedEmbeddings(nn.Module):
        super().__init__()
        self.height = height
        self.width = width
-        self.channel = channel
+        self.channel = (
+            channel  # TODO(fracapuano): this gives issues with non-square images bc is hardcoded to 4
+        )
        self.num_features = num_features

        self.kernel = nn.Parameter(torch.empty(channel, height, width, num_features))