def __init__(self, num_bits=8, quant_delay=0, ema_decay=0.999, symmetric=False, narrow_range=False): super(HSwishQuant, self).__init__() self.fake_quant_act_before = 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.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.HSwish()
def __init__(self): super(HSwish, self).__init__() self.hswish = P.HSwish()
def __init__(self): super(Net, self).__init__() self.HSwish = P.HSwish()