From 500aa12318ce5acd289d5edb6cb8266b3c3b162e Mon Sep 17 00:00:00 2001 From: LysandreJik Date: Wed, 8 Apr 2020 14:32:47 -0400 Subject: [PATCH] close #3699 --- src/transformers/modeling_electra.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_electra.py b/src/transformers/modeling_electra.py index 85996cb6a1..2d46716c35 100644 --- 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