Method __init__

mom/models/mom/modeling_mom.py:199–200

(self, *inputs, **kwargs)

    _no_split_modules = ['MomBlock']

    def __init__(self, *inputs, **kwargs):
        super().__init__(*inputs, **kwargs)

    def _init_weights(
        self,