remove bnneck
parent
d7a20aa967
commit
e651f3f62f
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue