Update neighborexchange.py

neighborexchange.py

@@ -56,7 +56,7 @@ class NeighborExchange(nn.Module):
                 if i < self.num_experts_x - 1: # Down neighbor
                     neighbors.append(reshaped_outputs[:, :, i+1, j, :])
                 if j > 0: # Left neighbor
-                    neighbors.append(reshpaced_outputs[:, :, i, j-1, :])
+                    neighbors.append(reshaped_outputs[:, :, i, j-1, :])
                 if j < self.num_experts_y - 1: # Right neighbor
                     neighbors.append(reshaped_outputs[:, :, i, j+1, :])