From d9731a52cf5ee1a402dba9de84ec5e75a8a43930 Mon Sep 17 00:00:00 2001 From: lizhenyu <lizhenyu13@huawei.com> Date: Wed, 29 Jun 2022 17:56:47 +0800 Subject: [PATCH] remove rank table file in ps 1p script --- .../script/run_parameter_server_train_standalone.sh | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/official/recommend/wide_and_deep/script/run_parameter_server_train_standalone.sh b/official/recommend/wide_and_deep/script/run_parameter_server_train_standalone.sh index 62ee42365..a36e29cda 100644 --- a/official/recommend/wide_and_deep/script/run_parameter_server_train_standalone.sh +++ b/official/recommend/wide_and_deep/script/run_parameter_server_train_standalone.sh @@ -27,15 +27,14 @@ export MS_SERVER_NUM=$4 export MS_SCHED_HOST=$5 export MS_SCHED_PORT=$6 DEVICE_ID=$7 -export RANK_TABLE_FILE=$8 -export VOCAB_CACHE_SIZE=$9 -export SPARSE=${10} +export VOCAB_CACHE_SIZE=$8 +export SPARSE=$9 -if [[ ! -n "$9" ]]; then +if [[ ! -n "$8" ]]; then export VOCAB_CACHE_SIZE=0 fi -if [[ ! -n "${10}" ]]; then +if [[ ! -n "$9" ]]; then export SPARSE=False fi @@ -77,7 +76,6 @@ export MS_ROLE=MS_WORKER rm -rf ${execute_path}/worker/ mkdir ${execute_path}/worker/ cd ${execute_path}/worker/ || exit -export RANK_ID=0 python -s ${self_path}/../train_and_eval_parameter_server_standalone.py --device_target=$DEVICE_TARGET \ --epochs=$EPOCH_SIZE --data_path=$DATASET --parameter_server=1 \ --vocab_cache_size=$VOCAB_CACHE_SIZE --sparse=$SPARSE --dropout_flag=True >worker.log 2>&1 & -- GitLab