有简单手写的语义分割的项目可以参考吗?
收藏
采用Camvid数据集,enet网络进行语义分割。label类别数目为32,一训练就会出现下面的保错,是内存的问题吗,batch_size调的很小GPU使用根本到不了32G,然后将label=label/2就能运行,但label=label/2得到的int类型不就相当于label数目少了一半吗?有些疑惑,还是因为label不是32类?
0
收藏
请登录后评论
可能是显存超出,具体看看这篇文档debug试试
https://www.paddlepaddle.org.cn/documentation/docs/zh/advanced_guide/performance_improving/device_switching/device_switching.html
可以看看cpu,gpu运行状态
运行时脚本页面下边的状态栏有按钮,点出选项卡看
您说的显存是这个吗?,
最近训练时会出现V100的显存不足的问题
图像不用、img=img/255.0都能运行,label非得除点啥才能运行!!!!这个保错到底是什么意思
我记得Camvid数据集是32类语义标签。。应该不是显存问题,这个只用了11G左右