游雁 2 лет назад
Родитель
Сommit
6c8e839742
2 измененных файлов с 4 добавлено и 8 удалено
  1. 3 8
      examples/aishell/conformer/run.sh
  2. 1 0
      funasr/models/transformer/model.py

+ 3 - 8
examples/aishell/conformer/run.sh

@@ -109,14 +109,9 @@ if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
   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}" \

+ 1 - 0
funasr/models/transformer/model.py

@@ -118,6 +118,7 @@ class Transformer(nn.Module):
         #         token_list, sym_space, sym_blank, report_cer, report_wer
         #     )
         #
+        self.error_calculator = None
         if ctc_weight == 0.0:
             self.ctc = None
         else: