脚本执行

准备数据集

准备数据集并上传到运行环境的目录下,例如:/home/data/resnet50/imagenet。

准备ranktable文件

ranktable文件样例和文件说明请参考准备资源配置文件

配置环境变量

请参考执行单Device训练配置环境变量。

执行命令

python3 /home/official/r1/resnet/imagenet_main.py --batch_size=32 --hooks=ExamplesPerSecondHook --data_dir=/home/data/resnet50/imagenet