mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

remove bnneck

Browse Source
This commit is contained in:
weishengyu 2022-04-08 10:58:23 +08:00
parent d7a20aa967
commit e651f3f62f
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/arch/gears/bnneck.py
View File

@ -6,7 +6,7 @@ class BNNeck(paddle.nn.Layer):
super(BNNeck, self).__init__() super(BNNeck, self).__init__()
self.num_filters = num_filters self.num_filters = num_filters
self.bn = paddle.nn.BatchNorm( self.bn = paddle.nn.BatchNorm1D(
self.num_filters) self.num_filters)
if not trainable: if not trainable:
self.bn.weight.trainable = False self.bn.weight.trainable = False

8
ppcls/configs/Pedestrian/strong_baseline_m1.yaml
View File

@ -26,10 +26,10 @@ Arch:
stem_act: null stem_act: null
BackboneStopLayer: BackboneStopLayer:
name: "flatten" name: "flatten"
Neck: #Neck:
name: BNNeck # name: BNNeck
num_filters: 2048 # num_filters: 2048
trainable: false # trainable: false
Head: Head:
name: "FC" name: "FC"
embedding_size: 2048 embedding_size: 2048