Skip to content

GitLab

Commit 6448396d authored by thomwolf's avatar thomwolf
Browse files

fix roberta test

parent 1e47dee2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_transformers/modeling_tf_roberta.py
View file @ 6448396d
...@@ -351,7 +351,7 @@ class TFRobertaForSequenceClassification(TFRobertaPreTrainedModel): ...@@ -351,7 +351,7 @@ class TFRobertaForSequenceClassification(TFRobertaPreTrainedModel):
outputs = self.roberta(inputs, **kwargs) outputs = self.roberta(inputs, **kwargs)
sequence_output = outputs[0] sequence_output = outputs[0]
logits = self.classifier(sequence_output, training=training) logits = self.classifier(sequence_output, training=kwargs.get('training', False))
outputs = (logits,) + outputs[2:] outputs = (logits,) + outputs[2:]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment