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()
