Fix wrong projection output

khalilsnoussi · May 1, 2024 · ea38b95 · ea38b95
1 parent 68e6447
commit ea38b95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_mamba/model.py b/vision_mamba/model.py
@@ -95,7 +95,7 @@ def forward(self, x: torch.Tensor):
 
         # Split x into x1 and x2 with linears
         z1 = self.proj(x)
-        x1 = self.proj(x)
+        x = self.proj(x)
 
         # forward con1d
         x1 = self.process_direction(