close #3699

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