From 33f94e3d917a54715af8cecdfcb9f845da24897e Mon Sep 17 00:00:00 2001 From: "Li-Huai (Allan) Lin" Date: Mon, 29 Aug 2022 18:15:34 +0800 Subject: [PATCH] up --- src/transformers/models/luke/modeling_luke.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/luke/modeling_luke.py b/src/transformers/models/luke/modeling_luke.py index cca68cd5354bc3..d54a9e7c344917 100644 --- a/src/transformers/models/luke/modeling_luke.py +++ b/src/transformers/models/luke/modeling_luke.py @@ -224,7 +224,7 @@ class EntitySpanClassificationOutput(ModelOutput): Args: loss (`torch.FloatTensor` of shape `(1,)`, *optional*, returned when `labels` is provided): Classification loss. - logits (`torch.FloatTensor` of shape `(batch_size, config.num_labels)`): + logits (`torch.FloatTensor` of shape `(batch_size, entity_length, config.num_labels)`): Classification scores (before SoftMax). hidden_states (`tuple(torch.FloatTensor)`, *optional*, returned when `output_hidden_states=True` is passed or when `config.output_hidden_states=True`): Tuple of `torch.FloatTensor` (one for the output of the embeddings + one for the output of each layer) of