训练过程中报错

[zxl222]

您好，感谢您的代码，这对我很有用。
在训练了几个epoch后，出现错误：RunTimeError: stack expects each tensor to be equal size. but got [256] at entry 0 and [345] at entry 3.
我实在找不到问题的原因，您知道这是什么问题导致的吗。
期待收到您的回复
@WuHuRestaurant