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()
def __init__(self): super(HSigmoid, self).__init__() self.hsigmoid = P.HSigmoid()
def __init__(self): super(DynamicNet, self).__init__() self.HSigmoid = P.HSigmoid() self.d = inner.GpuConvertToDynamicShape()
def __init__(self): super(Net, self).__init__() self.HSigmoid = P.HSigmoid()