Browse Source

refine wide and deep model zoo log file

tags/v1.1.0
lizhenyu 5 years ago
parent
commit
a41ec4668b
2 changed files with 4 additions and 4 deletions
  1. +3
    -3
      model_zoo/official/recommend/wide_and_deep/script/run_parameter_server_train_cluster.sh
  2. +1
    -1
      model_zoo/official/recommend/wide_and_deep/script/run_parameter_server_train_distribute.sh

+ 3
- 3
model_zoo/official/recommend/wide_and_deep/script/run_parameter_server_train_cluster.sh View File

@@ -70,9 +70,9 @@ if [[ "$MS_ROLE" == "MS_WORKER" ]]; then
rm -rf ${execute_path}/worker/
mkdir ${execute_path}/worker/
cd ${execute_path}/worker/ || exit
mpirun --allow-run-as-root -n $LOCAL_WORKER_NUM \
python -s ${self_path}/../train_and_eval_parameter_server_distribute.py \
--device_target=$DEVICE --data_path=$DATASET --epochs=$EPOCH_SIZE --parameter_server=1 \
mpirun --allow-run-as-root -n $LOCAL_WORKER_NUM --output-filename log_output --merge-stderr-to-stdout \
python -s ${self_path}/../train_and_eval_parameter_server_distribute.py \
--device_target=$DEVICE --data_path=$DATASET --epochs=$EPOCH_SIZE --parameter_server=1 \
--vocab_cache_size=$VOCAB_CACHE_SIZE --dropout_flag=1 >worker.log 2>&1 &
else
for((i=0;i<$LOCAL_WORKER_NUM;i++));


+ 1
- 1
model_zoo/official/recommend/wide_and_deep/script/run_parameter_server_train_distribute.sh View File

@@ -62,7 +62,7 @@ if [[ "X$DEVICE_TARGET" == "XGPU" ]]; then
rm -rf ${execute_path}/worker/
mkdir ${execute_path}/worker/
cd ${execute_path}/worker/ || exit
mpirun --allow-run-as-root -n $RANK_SIZE \
mpirun --allow-run-as-root -n $RANK_SIZE --output-filename log_output --merge-stderr-to-stdout \
python -s ${self_path}/../train_and_eval_parameter_server_distribute.py \
--device_target=$DEVICE_TARGET --data_path=$DATASET --epochs=$EPOCH_SIZE --parameter_server=1 \
--vocab_cache_size=$VOCAB_CACHE_SIZE --dropout_flag=1 >worker.log 2>&1 &


Loading…
Cancel
Save