其他分享
首页 > 其他分享> > 训练分割网络时出现CUDA error: an illegal memory access was encountered

训练分割网络时出现CUDA error: an illegal memory access was encountered

作者:互联网

奇怪的问题,我发现VOC2012数据集中有20类标签,label除了1-20之外还有0表示背景,255表示边缘,想能不能把这两个也分割出来,所以设定DeepLab的类别数为22,结果在val的时候出现这个问题,当把类别设置为20时,则正常运行。 

标签:分割,20,22,illegal,encountered,memory,类别,was,255
来源: https://www.cnblogs.com/legendongary/p/14250971.html