diff --git a/model_zoo/official/nlp/transformer/scripts/run_distribute_train_ascend.sh b/model_zoo/official/nlp/transformer/scripts/run_distribute_train_ascend.sh index 057e93a0e56..b4a4c8fc228 100644 --- a/model_zoo/official/nlp/transformer/scripts/run_distribute_train_ascend.sh +++ b/model_zoo/official/nlp/transformer/scripts/run_distribute_train_ascend.sh @@ -28,6 +28,7 @@ cd run_distribute_train || exit EPOCH_SIZE=$2 DATA_PATH=$3 +export HCCL_CONNECT_TIMEOUT=600 export RANK_TABLE_FILE=$4 export RANK_SIZE=$1 export HCCL_FLAG=1