这个错误整了我一天时间所以特此记录一下,如果大家也碰到希望能够避雷!!!
这个错误是在运行PyTorch代码出错我碰到的原因是神经网络init定义时给的参数出错。
【1】网仩有人说这是gpu坏了不能运行,首先可以通过:
测试就可以排除这种问题。所一般不是cuda或者机器的问题
注:对于报错不要心存侥幸,,我开始就觉得是不是cudagpu抽风了,,所以一定要检查自己的代码!!!
【2】如果torch能够正常运行说明设备没有问题,那么肯定是你的玳码出错了!!!!!!
由于是cuda运行错误所以报错位置可能不准确。建议大家一行行排查
这个错误是由于索引越界产生的。
coco数据集就囿这个问题需要重新将类别id(cat_ids)从零开始。
传入的vocab_size数比你词典中个数少
ps:遇到这种错误好好检查代码欢迎解决问题得留下容易错误的原因。