From a2367a92b946e66c384c492f3cdde60988342832 Mon Sep 17 00:00:00 2001
From: weishengyu <weishengyu@baidu.com>
Date: Tue, 25 May 2021 23:43:41 +0800
Subject: [PATCH] dbg

---
 ppcls/arch/backbone/legendary_models/hrnet.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/ppcls/arch/backbone/legendary_models/hrnet.py b/ppcls/arch/backbone/legendary_models/hrnet.py
index 20cd01db..a36f3ee4 100644
--- a/ppcls/arch/backbone/legendary_models/hrnet.py
+++ b/ppcls/arch/backbone/legendary_models/hrnet.py
@@ -582,15 +582,17 @@ class HRNet(TheseusLayer):
 
         la1 = self.la1(conv2)
 
-        tr1 = self.tr1_1([la1])
-        tr1 = self.tr1_2(tr1)
-        st2 = self.st2(tr1)
+        tr1_1 = self.tr1_1([la1])
+        tr1_2 = self.tr1_2([la1])
+        st2 = self.st2([tr1_1, tr1_2])
 
-        tr2 = self.tr2(st2)
-        st3 = self.st3(tr2)
+        tr2 = self.tr2(st2[-1])
+        st2.append(tr2)
+        st3 = self.st3(st2)
 
-        tr3 = self.tr3(st3)
-        st4 = self.st4(tr3)
+        tr3 = self.tr3(st3[-1])
+        st3.append(tr3)
+        st4 = self.st4(st3)
 
         last_cls = self.last_cls(st4)
 
-- 
GitLab