update-MLCA

ucas-dx · Dec 1, 2023 · 5ef0d5a · 5ef0d5a
1 parent 332f741
commit 5ef0d5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cv-attention/MLCA.py b/cv-attention/MLCA.py
@@ -38,7 +38,7 @@ def forward(self, x):
 
         # (b,c,local_size,local_size) <- (b,c,local_size*local_size)<-(b,local_size*local_size,c) <- (b,1,local_size*local_size*c)
         y_local_transpose=y_local.reshape(b, self.local_size * self.local_size,c).transpose(-1,-2).view(b, c, self.local_size , self.local_size)
-        # (b,1,c) -> (b,c) -> (c,b,1,1)
+        # (b,1,c) -> (b,c,1) -> (c,b,1,1)
         y_global_transpose = y_global.transpose(-1,-2).unsqueeze(-1)
 
         # 反池化