模型库DeepLabV3+,eval.py里面有mean_iou遇到问题,GPU运行不了
Created by: Exception-star
-
版本、环境信息:
1)PaddlePaddle版本:1.5.2
2)CPU:i9-9900k
3)GPU:RTX-2080Ti
4)系统环境:win10
1)模型名称 DeepLabV3+
-
复现信息:程序运行到箭头所指地方,程序无缘无故退出
-
问题描述:
是不是张量(label)在GPU,标量(num_classes)在CPU上无法比较,导致程序退出。
在Place =CPU可以比较,Place =GPU程序会退出