p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify weight offset for air

This commit is contained in:
zhang__sss 2021-06-17 19:40:38 +08:00
parent 494639ad8e
commit 40ef894029
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/compression/export/quant_export.py
View File

@ -85,7 +85,7 @@ class QuantBlock(Cell):
self.has_act = activation is not None
self.bias_add = P.BiasAdd()
self.sub = P.Sub()
self.weight_offset = Parameter(np.zeros(shape=weight.shape, dtype=np.int8), name='weight_offset')
self.weight_offset = Parameter(np.zeros(1, dtype=np.int8), name='weight_offset')
def construct(self, x):
x = self.quant(x)