提交 ad3c8d6b 编写于 作者: Z Ziyan

fix typo in distributed training script

上级 73aed007
...@@ -275,7 +275,7 @@ def test_train_cifar(num_classes=10, epoch_size=10): ...@@ -275,7 +275,7 @@ def test_train_cifar(num_classes=10, epoch_size=10):
```bash ```bash
#!/bin/bash #!/bin/bash
DATD_PATH=$1 DATA_PATH=$1
export DATA_PATH=${DATA_PATH} export DATA_PATH=${DATA_PATH}
RANK_SIZE=$2 RANK_SIZE=$2
......
#!/bin/bash #!/bin/bash
DATD_PATH=$1 DATA_PATH=$1
export DATA_PATH=${DATA_PATH} export DATA_PATH=${DATA_PATH}
RANK_SIZE=$2 RANK_SIZE=$2
EXEC_PATH=$(pwd) 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 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 export RANK_SIZE=2
} }
test_dist_${RANK_SIZE}p test_dist_${RANK_SIZE}pcs
for((i=0;i<${RANK_SIZE};i++)) for((i=0;i<${RANK_SIZE};i++))
do do
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册