def _reset_parameters(self):
        r"""Initiate parameters in the transformer model."""

        for p in self.parameters():
            if p.dim() > 1:
                xavier_uniform_(p)
Initiate parameters in the transformer model.
analyze/convnexts4nd/src/models/baselines/transformer.py