提交 e651f3f6 编写于 作者: W weishengyu

remove bnneck

上级 d7a20aa9
......@@ -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
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册