close #3699

2020-04-08 14:32:47 -04:00
parent a594ee9c84
commit 500aa12318
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/modeling_electra.py
+++ b/src/transformers/modeling_electra.py
@@ -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