Ejemplo n.º 1
0
 def __init__(self,
              num_bits=8,
              quant_delay=0,
              ema_decay=0.999,
              symmetric=False,
              narrow_range=False):
     super(HSigmoidQuant, self).__init__()
     self.fake_quant_act_before = FakeQuantWithMinMax(
         min_init=-6,
         max_init=6,
         num_bits=num_bits,
         quant_delay=quant_delay,
         ema=True,
         symmetric=symmetric,
         narrow_range=narrow_range)
     self.fake_quant_act_after = FakeQuantWithMinMax(
         min_init=-6,
         max_init=6,
         num_bits=num_bits,
         quant_delay=quant_delay,
         ema=True,
         ema_decay=ema_decay,
         symmetric=symmetric,
         narrow_range=narrow_range)
     self.act = P.HSigmoid()
Ejemplo n.º 2
0
 def __init__(self):
     super(HSigmoid, self).__init__()
     self.hsigmoid = P.HSigmoid()
Ejemplo n.º 3
0
 def __init__(self):
     super(DynamicNet, self).__init__()
     self.HSigmoid = P.HSigmoid()
     self.d = inner.GpuConvertToDynamicShape()
Ejemplo n.º 4
0
 def __init__(self):
     super(Net, self).__init__()
     self.HSigmoid = P.HSigmoid()