SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix next sentence output (#8466)

Browse Source
This commit is contained in:
Julien Plu
2020-11-11 15:41:39 +01:00
committed by GitHub
parent da842e4e72
commit 069b63844c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_tf_outputs.py
View File

@@ -325,7 +325,7 @@ class TFNextSentencePredictorOutput(ModelOutput):
heads. heads.
""" """
loss: tf.Tensor = None loss: Optional[tf.Tensor] = None
logits: tf.Tensor = None logits: tf.Tensor = None
hidden_states: Optional[Tuple[tf.Tensor]] = None hidden_states: Optional[Tuple[tf.Tensor]] = None
attentions: Optional[Tuple[tf.Tensor]] = None attentions: Optional[Tuple[tf.Tensor]] = None