dbg

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

Showing with 2 addition and 1 deletion

ppcls/arch/backbone/legendary_models/hrnet.py ppcls/arch/backbone/legendary_models/hrnet.py +2 -1

未找到文件。
--- a/ppcls/arch/backbone/legendary_models/hrnet.py
+++ b/ppcls/arch/backbone/legendary_models/hrnet.py
@@ -582,7 +582,8 @@ class HRNet(TheseusLayer):

        la1 = self.la1(conv2)

-        tr1 = self.tr1([la1])
+        tr1 = self.tr1_1([la1])
+        tr1 = self.tr1_2(tr1)
        st2 = self.st2(tr1)

        tr2 = self.tr2(st2)