FlaubertForSequenceClassification inherit from FlaubertModel instead …

…of XLMPreTrainedModel
ThilinaRajapakse · Feb 10, 2020 · 0ae8834 · 0ae8834
1 parent 048b3eb
commit 0ae8834
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simpletransformers/classification/transformer_models/flaubert_model.py b/simpletransformers/classification/transformer_models/flaubert_model.py
@@ -5,7 +5,7 @@
 from torch.nn import CrossEntropyLoss, MSELoss
 
 
-class FlaubertForSequenceClassification(XLMPreTrainedModel):
+class FlaubertForSequenceClassification(FlaubertModel):
     r"""
         **labels**: (`optional`) ``torch.LongTensor`` of shape ``(batch_size,)``:
             Labels for computing the sequence classification/regression loss.