From 42df92b8b1a25b91623091675bccc63d19b4ef16 Mon Sep 17 00:00:00 2001 From: mao_yin324 <254473743@qq.com> Date: Thu, 24 Mar 2022 03:09:12 +0000 Subject: [PATCH] update models.py. --- .../image_classification/PASSRnet_ID0986_for_PyTorch/models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/PyTorch/dev/cv/image_classification/PASSRnet_ID0986_for_PyTorch/models.py b/PyTorch/dev/cv/image_classification/PASSRnet_ID0986_for_PyTorch/models.py index 7782107046..e3c5191a5f 100644 --- a/PyTorch/dev/cv/image_classification/PASSRnet_ID0986_for_PyTorch/models.py +++ b/PyTorch/dev/cv/image_classification/PASSRnet_ID0986_for_PyTorch/models.py @@ -181,6 +181,7 @@ class PAM(nn.Module): ### fusion buffer = self.b3(x_right).permute(0,2,3,1).contiguous().view(-1, w, c) # (B*H) * W * C buffer = torch.bmm(M_right_to_left, buffer).contiguous().view(b, h, w, c).permute(0,3,1,2) # B * C * H * W + V_left_to_right = V_left_to_right.to(x_left.dtype) out = self.fusion(torch.cat((buffer, x_left, V_left_to_right), 1)) ## output -- Gitee