SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Change param order for consistency

Browse Source
This commit is contained in:
Julien Chaumond
2019-11-26 12:54:36 -05:00
committed by Lysandre Debut
parent c536c2a480
commit bdfe21ab24
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
transformers/modeling_albert.py
View File

@@ -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,