Richard Zowalla created OPENNLP-1979:
----------------------------------------
Summary: SentenceVectorsDL fails with a ClassCastException on
models whose first output is a pooled [batch, hidden] tensor
Key: OPENNLP-1979
URL: https://issues.apache.org/jira/browse/OPENNLP-1979
Project: OpenNLP
Issue Type: Bug
Reporter: Richard Zowalla
Fix For: 3.0.0
{{SentenceVectorsDL}} casts the first model output to {{float\[\]\[\]\[\]}}
\({{\[batch, sequence, hidden\]}}\). Many sentence\-transformers ONNX exports
expose {{sentence\_embedding}} with shape {{\[batch, hidden\]}} as the first
output, which makes the cast throw {{ClassCastException}}. The output should be
selected by name and by rank, and a pooled 2\-D output used as it is.
--
This message was sent by Atlassian Jira
(v8.20.10#820010)
