diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_4p.sh b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_4p.sh index be35617020cbb16f85e21961533089b5dd0d6c51..8296478223a22d687cde6e67747bab12d72d0946 100644 --- a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_4p.sh +++ b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_4p.sh @@ -5,7 +5,7 @@ cur_path=`pwd` #export ASCEND_SLOG_PRINT_TO_STDOUT=1 #export GE_USE_STATIC_MEMORY=1 -export HCCL_CONNECT_TIMEOUT=300 +export HCCL_CONNECT_TIMEOUT=1200 #集合通信参数,不需要修改 export RANK_SIZE=4 @@ -24,8 +24,8 @@ RankSize=1 #参数配置 data_path="/npu/traindata/ID2940_CarPeting_TF_WideDeep_TF" -train_size=13107200 -display_step=1 +train_size=52428800 +display_step=10 n_epoches=4 #维持参数,以下不需要修改 diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_8p.sh b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_8p.sh index 55f195dd29fb41a89944b3bd9a63fc3663d1c2c5..bb0e0a8447cf22315bfaecd5e471ff866e14643e 100644 --- a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_8p.sh +++ b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/test/train_performance_8p.sh @@ -5,7 +5,7 @@ cur_path=`pwd` #export ASCEND_SLOG_PRINT_TO_STDOUT=1 #export GE_USE_STATIC_MEMORY=1 -export HCCL_CONNECT_TIMEOUT=300 +export HCCL_CONNECT_TIMEOUT=1200 #集合通信参数,不需要修改 export RANK_SIZE=8 @@ -24,8 +24,8 @@ RankSize=1 #参数配置 data_path="/npu/traindata/ID2940_CarPeting_TF_WideDeep_TF" -train_size=13107200 -display_step=1 +train_size=52428800 +display_step=10 n_epoches=4 #维持参数,以下不需要修改