Method __init__

mogen/models/rnns/t2m_bigru.py:270–275

(self, motion_encoder=None, text_encoder=None, init_cfg=None)

class T2MContrastiveModel(BaseModule):

    def __init__(self, motion_encoder=None, text_encoder=None, init_cfg=None):
        super().__init__()
        self.motion_encoder = T2MMotionEncoder(**motion_encoder)
        self.text_encoder = T2MTextEncoder(**text_encoder)
        assert init_cfg['type'] == 'Pretrained'
        self.load_pretrained(init_cfg['checkpoint'])

    def encode_motion(self,
                      motion,