update

游雁

2024-02-20 58b6154a73331a8807127d4579ed473432ce88de

 examples/aishell/conformer/run.sh

@@ -108,15 +108,10 @@
  log_file="${exp_dir}/exp/${model_dir}/train.log.txt"
  echo "log_file: ${log_file}"

  gpu_num=$(echo CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
#  torchrun \
#  --nnodes 1 \
#  --nproc_per_node ${gpu_num}
  cmd="python"
  if [ ${gpu_num} -gt 1  ];then
    cmd="torchrun --nnodes 1 --nproc_per_node ${gpu_num}"
  fi
  ${cmd} \
  gpu_num=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
  torchrun \
  --nnodes 1 \
  --nproc_per_node ${gpu_num} \
  ../../../funasr/bin/train.py \
  --config-path "${workspace}/conf" \
  --config-name "${config}" \