diff --git a/ppcls/arch/backbone/legendary_models/hrnet.py b/ppcls/arch/backbone/legendary_models/hrnet.py
index d720fdfee..20cd01db8 100644
--- 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)