准备数据集:
- 首先下载数据集
wget http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz
- 然后解压数据集
tar -zxvf cifar-10-binary.tar.gz
下载代码并简单修改训练脚本:
- 在MindSpore的docs仓下载代码
https://gitee.com/mindspore/docs/tree/r1.2/tutorials/tutorial_code/resnet
- 修改训练设备
device_target="CPU"
- 修改数据集路径
parser.add_argument('--dataset_path', type=str, default="\***\cifar-10-batches-bin", help='Dataset path.')
执行训练
python main.py
查看训练loss输出