首页 Paddle Lite 帖子详情
GPU训练没有CPU训练出来的数据集预测精准
收藏
快速回复
Paddle Lite 问答推理部署 595 1
GPU训练没有CPU训练出来的数据集预测精准
收藏
快速回复
Paddle Lite 问答推理部署 595 1

训练自己的VOC数据集,使用GPU训练的没有CPU训练下的结果预测精准是怎么回事呢?使用的是同一个模型文件,同一个数据集文件,GPU训练出来的识别box特别大,可能框选整个图片,CPU模式下训练出来的识别框选很精准,是同一个环境,只是通过模型文件.yml里面的参数use_gpu: false  改成true或者false去训练?

1
收藏
回复
全部评论(1)
时间顺序
雷青
#2 回复于2021-11

你确定GPU训练了吗?可以看下GPU和CPU训练过程中的loss下降曲线,首先确认GPU训练没问题。再有问题,可以在PaddlePaddle的github上提issue。https://github.com/PaddlePaddle/Paddle/issues

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户