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