new

语帆

2024-02-21 eeccdc9a5d72f496f5e7b2a0e3dd381bebcc6ff9

 examples/aishell/paraformer/run.sh

@@ -105,10 +105,11 @@
  echo "stage 4: ASR Training"

  mkdir -p ${exp_dir}/exp/${model_dir}
  log_file="${exp_dir}/exp/${model_dir}/train.log.txt"
  current_time=$(date "+%Y-%m-%d_%H-%M")
  log_file="${exp_dir}/exp/${model_dir}/train.log.txt.${current_time}"
  echo "log_file: ${log_file}"

  gpu_num=$(echo CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
  gpu_num=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
  torchrun \
  --nnodes 1 \
  --nproc_per_node ${gpu_num} \