From 15884fe78fc40872e184cd134fac8291453156a4 Mon Sep 17 00:00:00 2001
From: weishengyu <weishengyu@baidu.com>
Date: Tue, 25 May 2021 23:58:51 +0800
Subject: [PATCH] remove layer1

---
 ppcls/arch/backbone/legendary_models/hrnet.py | 29 ++++++-------------
 1 file changed, 9 insertions(+), 20 deletions(-)

diff --git a/ppcls/arch/backbone/legendary_models/hrnet.py b/ppcls/arch/backbone/legendary_models/hrnet.py
index fdbce664..b37b953f 100644
--- a/ppcls/arch/backbone/legendary_models/hrnet.py
+++ b/ppcls/arch/backbone/legendary_models/hrnet.py
@@ -75,25 +75,6 @@ class ConvBNLayer(TheseusLayer):
         return y
 
 
-class Layer1(TheseusLayer):
-    def __init__(self, num_channels, has_se=False, name=None):
-        super(Layer1, self).__init__()
-
-        self.bottleneck_blocks = nn.Sequential(*[BottleneckBlock(
-                num_channels=num_channels if i == 0 else 256,
-                num_filters=64,
-                has_se=has_se,
-                stride=1,
-                downsample=True if i == 0 else False,
-                name=name + '_' + str(i + 1))
-            for i in range(4)
-        ])
-
-    def forward(self, x, res_dict=None):
-        y = self.bottleneck_blocks(x)
-        return y
-
-
 class Branches(TheseusLayer):
     def __init__(self,
                  block_num,
@@ -494,7 +475,15 @@ class HRNet(TheseusLayer):
             stride=2,
             act='relu')
 
-        self.la1 = Layer1(num_channels=64, has_se=has_se, name="layer2")
+        self.la1 = self.bottleneck_blocks = nn.Sequential(*[BottleneckBlock(
+                num_channels=64 if i == 0 else 256,
+                num_filters=64,
+                has_se=has_se,
+                stride=1,
+                downsample=True if i == 0 else False,
+                name="layer2_{}".format(i+1))
+            for i in range(4)
+        ])
 
         self.tr1_1 = ConvBNLayer(
             num_channels=256,
-- 
GitLab