脚本执行
准备数据集
准备数据集并上传到运行环境的目录下,例如:/home/data/resnet50/imagenet。
准备ranktable文件
ranktable文件样例和文件说明请参考准备ranktable资源配置文件。
配置环境变量
请参考执行单Device训练配置环境变量。
执行命令
python3 /home/official/r1/resnet/imagenet_main.py --batch_size=32 --hooks=ExamplesPerSecondHook --data_dir=/home/data/resnet50/imagenet
父主题: 手工迁移样例(ResNet50模型)