diff --git a/ppcls/arch/backbone/legendary_models/hrnet.py b/ppcls/arch/backbone/legendary_models/hrnet.py
index 41963660cff42be2cecf4182e211f48c29c7a2b9..ae1cc773d3fad2c533b4d1086d58f0042a9438be 100644
--- a/ppcls/arch/backbone/legendary_models/hrnet.py
+++ b/ppcls/arch/backbone/legendary_models/hrnet.py
@@ -496,16 +496,14 @@ class HRNet(TheseusLayer):
 
         self.la1 = Layer1(num_channels=64, has_se=has_se, name="layer2")
 
-        self.tr1_1 = BasicBlock(
+        self.tr1_1 = ConvBNLayer(
             num_channels=256,
             num_filters=width,
-            has_se=has_se,
-            name="tr1_1")
-        self.tr1_2 = BasicBlock(
+            filter_size=3)
+        self.tr1_2 = ConvBNLayer(
             num_channels=256,
             num_filters=width * 2,
-            has_se=has_se,
-            name="tr1_2")
+            filter_size=3)
 
         self.st2 = Stage(
             num_channels=channels_2,
@@ -514,11 +512,10 @@ class HRNet(TheseusLayer):
             has_se=self.has_se,
             name="st2")
 
-        self.tr2 = BasicBlock(
+        self.tr2 = ConvBNLayer(
             num_channels=width * 2,
             num_filters=width * 4,
-            has_se=has_se,
-            name="tr2")
+            filter_size=3)
         self.st3 = Stage(
             num_channels=channels_3,
             num_modules=num_modules_3,
@@ -526,11 +523,10 @@ class HRNet(TheseusLayer):
             has_se=self.has_se,
             name="st3")
 
-        self.tr3 = BasicBlock(
+        self.tr3 = ConvBNLayer(
             num_channels=width * 4,
             num_filters=width * 8,
-            has_se=has_se,
-            name="tr3")
+            filter_size=3)
 
         self.st4 = Stage(
             num_channels=channels_4,