模型有几个输出?

#7
by Jeromeblue - opened

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

Beijing Academy of Artificial Intelligence org

Sign up or log in to comment