Method load_cp

LanguageNetwork/BERT/models/model_builder.py:89–90

(self, pt)

        self.to(device)

    def load_cp(self, pt):
        self.load_state_dict(pt['models'], strict=True)

    def forward(self, x, segs, clss, mask, mask_cls, sentence_range=None):
        # print(x[0], segs[0], mask[0])