Change param order for consistency
This commit is contained in:
parent
c536c2a480
commit
bdfe21ab24
|
@ -571,8 +571,8 @@ class AlbertForMaskedLM(AlbertPreTrainedModel):
|
||||||
def get_output_embeddings(self):
|
def get_output_embeddings(self):
|
||||||
return self.predictions.decoder
|
return self.predictions.decoder
|
||||||
|
|
||||||
def forward(self, input_ids=None, attention_mask=None, token_type_ids=None, position_ids=None, head_mask=None,
|
def forward(self, input_ids=None, attention_mask=None, token_type_ids=None, position_ids=None, head_mask=None, inputs_embeds=None,
|
||||||
masked_lm_labels=None, inputs_embeds=None):
|
masked_lm_labels=None):
|
||||||
outputs = self.albert(
|
outputs = self.albert(
|
||||||
input_ids=input_ids,
|
input_ids=input_ids,
|
||||||
attention_mask=attention_mask,
|
attention_mask=attention_mask,
|
||||||
|
|
Loading…
Reference in New Issue