b3a133bd671f6a941380d33bf4980604a237bc80,nlp_architect/models/transformers/token_classification.py,XLNetForTokenClassification,__init__,#XLNetForTokenClassification#Any#,83

Before Change

class XLNetForTokenClassification(XLNetPreTrainedModel):
    def __init__(self, config):
        super(XLNetForTokenClassification, self).__init__(config)
        raise NotImplementedError

    def forward(self, input_ids, token_type_ids=None, input_mask=None, attention_mask=None,
                mems=None, perm_mask=None, target_mapping=None,
                labels=None, head_mask=None):

After Change

        self.transformer = XLNetModel(config)
        self.logits_proj = torch.nn.Linear(config.d_model, config.num_labels)
        self.dropout = torch.nn.Dropout(config.dropout)

        self.apply(self.init_weights)

    def forward(self, input_ids, token_type_ids=None, input_mask=None, attention_mask=None,

Instances