报错原因:

大概率是因为:代码有地方数组越界。盲猜是在交叉熵损失函数那里,反正我是在这里。

小概率是其他原因。不过,下面的解决办法是通用的。

解决办法:
先device=“cpu”,运行一遍,可以定位到数组越界的地方,修改代码。保证正确了之后再放在GPU上跑。

Logo

一站式 AI 云服务平台

更多推荐