def init_weights(self):
    for p in self.parameters():
        if (p.dim() > 1):
            nn.init.xavier_uniform_(p)
def init_weights(self):
    for p in self.parameters():
        if (p.dim() > 1):
            nn.init.xavier_uniform_(p)