Method __init__

diffplanner/models/rnns/t2m_bigru.py:108–122

(self,
                 word_size,
                 pos_size,
                 hidden_size,
                 output_size,
                 max_text_len)

class T2MTextEncoder(nn.Module):
    
    def __init__(self,
                 word_size,
                 pos_size,
                 hidden_size,
                 output_size,
                 max_text_len):
        super().__init__()
        self.text_encoder = TextEncoderBiGRUCo(
            word_size=word_size,
            pos_size=pos_size,
            hidden_size=hidden_size,
            output_size=output_size,
        )
        self.w_vectorizer = WordVectorizer('./data/glove', 'our_vab')
        self.max_text_len = max_text_len
        
    def load_pretrained(self, ckpt_path):
        checkpoint = torch.load(ckpt_path, map_location='cpu')