游雁
2024-02-20 6c8e8397424b02b6df3248824d24b9f993536f90
update
2个文件已修改
12 ■■■■■ 已修改文件
examples/aishell/conformer/run.sh 11 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/transformer/model.py 1 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
examples/aishell/conformer/run.sh
@@ -109,14 +109,9 @@
  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} \
  torchrun \
  --nnodes 1 \
  --nproc_per_node ${gpu_num} \
  ../../../funasr/bin/train.py \
  --config-path "${workspace}/conf" \
  --config-name "${config}" \
funasr/models/transformer/model.py
@@ -118,6 +118,7 @@
        #         token_list, sym_space, sym_blank, report_cer, report_wer
        #     )
        #
        self.error_calculator = None
        if ctc_weight == 0.0:
            self.ctc = None
        else: