def METHOD_NAME(self):
    if self.pooling_type == 'att':
        kaiming_init(self.conv_mask, mode='fan_in')
        self.conv_mask.inited = True
    if self.channel_add_conv is not None:
        last_zero_init(self.channel_add_conv)
    if self.channel_mul_conv is not None:
        last_zero_init(self.channel_mul_conv)