提交 e651f3f6 编写于 作者: W weishengyu

remove bnneck

上级 d7a20aa9
...@@ -6,7 +6,7 @@ class BNNeck(paddle.nn.Layer): ...@@ -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
......
...@@ -26,10 +26,10 @@ Arch: ...@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册