Hello, the author, after reading your code, found that there is an error ah, want to confirm, is the following code:lines 68 to 71 in the file
b, c, height, width = x.size()
C = self.pool(x)
H = self.pool(x.permute(0, 3, 1, 2).contiguous())
W = self.pool(x.permute(0, 2, 3, 1).contiguous())
H. W tensor definitions need to be exchanged?The dimension of the tensor is B,C,H,W.Thank you for your reply!
Hello, the author, after reading your code, found that there is an error ah, want to confirm, is the following code:lines 68 to 71 in the file b, c, height, width = x.size() C = self.pool(x) H = self.pool(x.permute(0, 3, 1, 2).contiguous()) W = self.pool(x.permute(0, 2, 3, 1).contiguous()) H. W tensor definitions need to be exchanged?The dimension of the tensor is B,C,H,W.Thank you for your reply!