Browse Source

modify weight offset for air

tags/v1.3.0
zhang__sss 4 years ago
parent
commit
40ef894029
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/compression/export/quant_export.py

+ 1
- 1
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)


Loading…
Cancel
Save