forked from huawei/mindspore2022
modify quant code
This commit is contained in:
parent
bc702459bc
commit
b97ed00c89
|
|
@ -205,7 +205,8 @@ class DenseBnAct(Cell):
|
|||
bias_init='zeros',
|
||||
has_bias=True,
|
||||
has_bn=False,
|
||||
activation=None):
|
||||
activation=None,
|
||||
after_fake=True):
|
||||
super(DenseBnAct, self).__init__()
|
||||
self.dense = basic.Dense(
|
||||
in_channels,
|
||||
|
|
@ -215,6 +216,7 @@ class DenseBnAct(Cell):
|
|||
has_bias)
|
||||
self.has_bn = validator.check_bool("has_bn", has_bn)
|
||||
self.has_act = activation is not None
|
||||
self.after_fake = after_fake
|
||||
if has_bn:
|
||||
self.batchnorm = BatchNorm1d(out_channels)
|
||||
self.activation = get_activation(activation)
|
||||
|
|
|
|||
Loading…
Reference in New Issue