def weight_init(self):
    for layer in self.linears:
        torch.nn.init.xavier_uniform(layer.weight)
        torch.nn.init.zeros_(layer.bias)
def weight_init(self):
    for layer in self.linears:
        torch.nn.init.xavier_uniform(layer.weight)
        torch.nn.init.zeros_(layer.bias)