报错
Pytorch: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMa
原因:
label不匹配
我自己是因为换了数据集,然后重新设置网络fc层的宽度,不小心设错了导致FC长度过小<Label总数量
导致调用criterion(crossEntropyLoss)的时候报错
查了好久才查出来 希望可以帮到有需要的人
报错
Pytorch: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMa
原因:
label不匹配
我自己是因为换了数据集,然后重新设置网络fc层的宽度,不小心设错了导致FC长度过小<Label总数量
导致调用criterion(crossEntropyLoss)的时候报错
查了好久才查出来 希望可以帮到有需要的人