diff --git a/tutorials/source_zh_cn/advanced_use/distributed_training.md b/tutorials/source_zh_cn/advanced_use/distributed_training.md index b7ded83975566aec6ada249397d00703e3cbfff1..1653348b0dc88ce9d283eed978fdf69e31700675 100644 --- a/tutorials/source_zh_cn/advanced_use/distributed_training.md +++ b/tutorials/source_zh_cn/advanced_use/distributed_training.md @@ -275,7 +275,7 @@ def test_train_cifar(num_classes=10, epoch_size=10): ```bash #!/bin/bash -DATD_PATH=$1 +DATA_PATH=$1 export DATA_PATH=${DATA_PATH} RANK_SIZE=$2 diff --git a/tutorials/tutorial_code/distributed_training/run.sh b/tutorials/tutorial_code/distributed_training/run.sh index 9d618ec767be1ad64cfae4c7aec3b8ba74683731..d8a58fb6653310ea4af07f0597d53b9ea4f61b06 100644 --- a/tutorials/tutorial_code/distributed_training/run.sh +++ b/tutorials/tutorial_code/distributed_training/run.sh @@ -1,24 +1,24 @@ #!/bin/bash -DATD_PATH=$1 +DATA_PATH=$1 export DATA_PATH=${DATA_PATH} RANK_SIZE=$2 EXEC_PATH=$(pwd) -test_dist_8p() +test_dist_8pcs() { - export MINDSPORE_HCCL_CONFIG_PATH=${EXEC_PATH}/rank_table_8p.json + export MINDSPORE_HCCL_CONFIG_PATH=${EXEC_PATH}/rank_table_8pcs.json export RANK_SIZE=8 } -test_dist_2p() +test_dist_2pcs() { - export MINDSPORE_HCCL_CONFIG_PATH=${EXEC_PATH}/rank_table_2p.json + export MINDSPORE_HCCL_CONFIG_PATH=${EXEC_PATH}/rank_table_2pcs.json export RANK_SIZE=2 } -test_dist_${RANK_SIZE}p +test_dist_${RANK_SIZE}pcs for((i=0;i<${RANK_SIZE};i++)) do