def METHOD_NAME(self):
    gain = nn.init.calculate_gain("relu")
    nn.init.xavier_normal_(self.fc.weight, gain=gain)
    nn.init.xavier_normal_(self.p, gain=gain)
    nn.init.xavier_normal_(self.attn_l, gain=gain)
    nn.init.xavier_normal_(self.attn_r, gain=gain)
    if self.residual:
        nn.init.xavier_normal_(self.residual_module.weight, gain=gain)