Browse Source

!4424 modify quant DenseBnAct API

Merge pull request !4424 from chengxb7532/r0.6
pull/4424/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
04a6612baf
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/quant/quant.py

+ 1
- 1
mindspore/train/quant/quant.py View File

@@ -274,7 +274,7 @@ class ConvertToQuantNetwork:
subcell.dense = dense_inner
if subcell.has_act and subcell.activation is not None:
subcell.activation = self._convert_activation(subcell.activation)
elif subcell.after_fake:
else:
subcell.has_act = True
subcell.activation = _AddFakeQuantAfterSubCell(F.identity,
num_bits=self.act_bits,


Loading…
Cancel
Save