dbg

2021-05-25 23:38:59 +08:00 · 2021-05-25 23:38:59 +08:00 · 4d4557a32d
parent 9ab8639c3d
commit 4d4557a32d
1 changed files with 2 additions and 1 deletions
--- 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)