Script Execution

Preparing a Dataset

Prepare a dataset and upload it to a directory in the operating environment, for example, /home/data/resnet50/imagenet.

Preparing the Ranktable File

For details about the ranktable file example and description, see "Reference" > "Cluster Information Configuration" in the HCCL User Guide.

Configuring Environment Variables

For details about the environment variable configuration, see Training with a Single Device.

Running the Command

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