Hugging Face's logo

BAAI
/
bge-base-zh

Feature Extraction
Transformers
PyTorch
Safetensors
Chinese
bert
text-embeddings-inference
Model card Files
xet
 Community
7

模型有几个输出？

#7
by Jeromeblue - opened
Discussion
Jeromeblue

用transformer bert进行forward时，报错提示预期有两个输出，但只得到一个。
start_logits, end_logits = logits.split(1, dim=-1)
ValueError: not enough values to unpack (expected 2, got 1)
Shitao
 Beijing Academy of Artificial Intelligence org

参考https://huggingface.co/BAAI/bge-base-zh#using-huggingface-transformers 输出embedding

Sign up or log in to comment