Closed Niki173 closed 2 years ago
triplet-attention.py
Hi,Run this file and find inconsistent dimensions
I wonder if there is a problem here
torch.cat( (torch.max(x, 1)[0].unsqueeze(1), torch.mean(x, 1).unsqueeze(1)), dim=1 )
it should be torch.cat( (torch.max(x, 1).unsqueeze(1), torch.mean(x, 1).unsqueeze(1)), dim=1 )
right?
@Niki173 apologies for the delay in response, for Z-pool, we follow the same code as that given in CBAM
triplet-attention.py
Hi,Run this file and find inconsistent dimensions
I wonder if there is a problem here
torch.cat( (torch.max(x, 1)[0].unsqueeze(1), torch.mean(x, 1).unsqueeze(1)), dim=1 )
it should be torch.cat( (torch.max(x, 1).unsqueeze(1), torch.mean(x, 1).unsqueeze(1)), dim=1 )
right?