pytorch报错:
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:224
原因:模型输出的类别个数和读入的类别个数不一致导致的
将输出类别个数修改为与输入类别个数一致即可。
pytorch报错:
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:224
原因:模型输出的类别个数和读入的类别个数不一致导致的
将输出类别个数修改为与输入类别个数一致即可。