from fastNLP.models import BaseModel from fastNLP.modules.encoder.bert import BertModel class BertForSNLI(BaseModel): # TODO: still in progress def __init(self): super(BertForSNLI, self).__init__() def forward(self, words, segment_id, seq_len): pass