问题现象描述

• 环境描述：
  • PyTorch 1.8.1，该问题不限版本
  • CANN：6.0.0.RC1/6.0.2/6.3.RC1，该问题不限版本
  • 产品型号：训练服务器
  • 代码路径：https://gitee.com/ascend/ModelZoo-PyTorch/tree/master/PyTorch/built-in/cv/classification/ResNet50_for_PyTorch
• 问题现象：

  PyTorch框架利用imagenet2012训练resnet50失败，报错“Couldn’t find any class folder in XXX”。

  放大

原因分析

分析之前能跑通的imagenet的数据集如下图所示：

放大

出错的val数据集：

放大

出错的train数据集：

放大

因此是数据集错误导致的。

解决措施

1. 制作符合要求的train数据集格式。
   1. 解压train压缩包，解压后是1000个tar包文件：tar –xvf ILSVRC2012_img_train.tar –C ./train。
   2. 编写脚本并运行，生成该代码要求的train数据集格式。

      放大

2. 制作符合要求的val数据集格式。
   1. 解压val压缩包，解压后是5000张图片：tar –xvf ILSVRC2012_img_val.tar –C ./val。
   2. 将val_prep.sh这个脚本放到val数据集中，和50000张图片放在一起，并运行。

      放大