for p in self.parameters():
            if len(p.shape) > 1:
                nn.init.xavier_uniform_(p)