在训练的时候报错,我个人遇到的是两个原因
1.标签数和最后一层网络的输出数量不一致
2.标签数没有转化成从0开始的n个数字(n对应类别)
比如预测5类,标签是0,1,2,3,4,注意,是从0开始
CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle) pytorch 错误总结
最新推荐文章于 2025-03-21 08:37:10 发布
在训练的时候报错,我个人遇到的是两个原因
1.标签数和最后一层网络的输出数量不一致
2.标签数没有转化成从0开始的n个数字(n对应类别)
比如预测5类,标签是0,1,2,3,4,注意,是从0开始