dbg

a2367a92 · weishengyu · 4d4557a3 · a2367a92
隐藏空白更改
内联并排

Showing with 9 addition and 7 deletion

ppcls/arch/backbone/legendary_models/hrnet.py ppcls/arch/backbone/legendary_models/hrnet.py +9 -7

未找到文件。
--- 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_1 = self.tr1_1([la1])
-        tr1 = self.tr1_2(tr1)
+        tr1_2 = self.tr1_2([la1])
-        st2 = self.st2(tr1)
+        st2 = self.st2([tr1_1, tr1_2])
-        tr2 = self.tr2(st2)
+        tr2 = self.tr2(st2[-1])
-        st3 = self.st3(tr2)
+        st2.append(tr2)
+        st3 = self.st3(st2)
-        tr3 = self.tr3(st3)
+        tr3 = self.tr3(st3[-1])
-        st4 = self.st4(tr3)
+        st3.append(tr3)
+        st4 = self.st4(st3)
        last_cls = self.last_cls(st4)