def reset_parameters(self):
    nn.init.zeros_(self.conv_gamma.weight)
    nn.init.zeros_(self.conv_beta.weight)
    nn.init.ones_(self.conv_gamma.bias)
    nn.init.zeros_(self.conv_beta.bias)
def reset_parameters(self):
    nn.init.zeros_(self.conv_gamma.weight)
    nn.init.zeros_(self.conv_beta.weight)
    nn.init.ones_(self.conv_gamma.bias)
    nn.init.zeros_(self.conv_beta.bias)