mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

dbg

pull/748/head
weishengyu 2021-05-25 23:55:12 +08:00
parent 1153ad6538
commit 94b1bae6ae
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ppcls/arch/backbone/legendary_models/hrnet.py
View File

@ -503,7 +503,9 @@ class HRNet(TheseusLayer):
self.tr1_2 = ConvBNLayer(
num_channels=256,
num_filters=width * 2,
filter_size=3)
filter_size=3,
stride=2
)
self.st2 = Stage(
num_channels=channels_2,
@ -515,7 +517,9 @@ class HRNet(TheseusLayer):
self.tr2 = ConvBNLayer(
num_channels=width * 2,
num_filters=width * 4,
filter_size=3)
filter_size=3,
stride=2
)
self.st3 = Stage(
num_channels=channels_3,
num_modules=num_modules_3,
@ -526,7 +530,9 @@ class HRNet(TheseusLayer):
self.tr3 = ConvBNLayer(
num_channels=width * 4,
num_filters=width * 8,
filter_size=3)
filter_size=3,
stride=2
)
self.st4 = Stage(
num_channels=channels_4,