dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

close #3699

This commit is contained in:
LysandreJik 2020-04-08 14:32:47 -04:00
parent a594ee9c84
commit 500aa12318
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_electra.py
View File

@ -662,7 +662,7 @@ class ElectraForTokenClassification(ElectraPreTrainedModel):
active_labels = labels.view(-1)[active_loss]
loss = loss_fct(active_logits, active_labels)
else:
loss = loss_fct(logits.view(-1, self.num_labels), labels.view(-1))
loss = loss_fct(logits.view(-1, self.config.num_labels), labels.view(-1))
output = (loss,) + output