!8897 fix QuantizationAwareTraining note typo error

From: @yuchaojie
Reviewed-by: @chenfei52,@sanjaychan
Signed-off-by: @sanjaychan
pull/8897/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit baf093ae35

@ -140,7 +140,7 @@ class QuantizationAwareTraining(Quantizer):
support QAT. Default: OptimizeOption.QAT
Examples:
>>> class Net(nn.Cell):
>>> class LeNet5(nn.Cell):
>>> def __init__(self, num_class=10, channel=1):
>>> super(LeNet5, self).__init__()
>>> self.type = "fusion"
@ -168,7 +168,7 @@ class QuantizationAwareTraining(Quantizer):
>>> x = self.fc3(x)
>>> return x
>>>
>>> net = Net()
>>> net = LeNet5()
>>> quantizer = QuantizationAwareTraining(bn_fold=False, per_channel=[True, False], symmetric=[True, False])
>>> net_qat = quantizer.quantize(net)
"""

Loading…
Cancel
Save