使用trans来进行推理时候出现的错误:

#4
by shams123321 - opened
value_layer = value_layer.view(output_size[0] * output_size[1], value_layer.size(2), -1)

RuntimeError: shape '[192, 6, -1]' is invalid for input of size 24576
是因为什么原因?

下面是代码部分:
image.png

transformer更新到了最新
image.png

zRzRzRzRzRzRzR changed discussion status to closed

Sign up or log in to comment