From c2a5b6fea4229ac4385b21796a4addc1eb1529e5 Mon Sep 17 00:00:00 2001
From: littleSpongebob <44308761+littleSpongebob@users.noreply.github.com>
Date: Fri, 23 Jul 2021 10:52:43 +0800
Subject: [PATCH] Update fusion.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

第二次全局注意力模块没有用到
---
 aff_pytorch/aff_net/fusion.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aff_pytorch/aff_net/fusion.py b/aff_pytorch/aff_net/fusion.py
index fef49b5..8585c26 100644
--- a/aff_pytorch/aff_net/fusion.py
+++ b/aff_pytorch/aff_net/fusion.py
@@ -71,7 +71,7 @@ def forward(self, x, residual):
         xi = x * wei + residual * (1 - wei)
 
         xl2 = self.local_att2(xi)
-        xg2 = self.global_att(xi)
+        xg2 = self.global_att2(xi)
         xlg2 = xl2 + xg2
         wei2 = self.sigmoid(xlg2)
         xo = x * wei2 + residual * (1 - wei2)