官网的github里图像分类代码无法正常训练
收藏
https://github.com/PaddlePaddle/models/tree/develop/fluid/PaddleCV/image_classification
以上地址的图像分类demo,无法正常训练,提示Imagnet数据集中的图片找不到,确认运行download_imagenet2012.sh,数据集都已经成功下载解压。对应的文件也存在的,但是程序运行就是提示找不到该文件。
0
收藏
请登录后评论
用的是数据集自己的测试集训练集reader吗
read.py里头把
img_path=img_path.replace("JPEG","jpeg")
改成
img_path=img_path.replace("jpeg","JPEG")
这个问题解决了,又运到新问题,错误如下图 ,我没有运行其他程序。

字面提示是显存不够了。。我也不清楚怎么调整了,可以去github上提个issue问问