remove bnneck

pull/1819/head
weishengyu 2022-04-08 10:58:23 +08:00
parent d7a20aa967
commit e651f3f62f
2 changed files with 5 additions and 5 deletions

View File

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

View File

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