def test(self, sents_tensor, lengths, _):
    logits = self.forward(sents_tensor, lengths)
    (_, batch_tagids) = torch.max(logits, dim=2)
    return batch_tagids
def test(self, sents_tensor, lengths, _):
    logits = self.forward(sents_tensor, lengths)
    (_, batch_tagids) = torch.max(logits, dim=2)
    return batch_tagids