speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
游雁 2024-02-20 15:53:06 +08:00
parent 362a6de3c8
commit 6c8e839742
2 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
examples/aishell/conformer/run.sh
View File

@ -109,14 +109,9 @@ if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
echo "log_file: ${log_file}" 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 \ torchrun \
# --nnodes 1 \ --nnodes 1 \
# --nproc_per_node ${gpu_num} --nproc_per_node ${gpu_num} \
cmd="python"
if [ ${gpu_num} -gt 1 ];then
cmd="torchrun --nnodes 1 --nproc_per_node ${gpu_num}"
fi
${cmd} \
../../../funasr/bin/train.py \ ../../../funasr/bin/train.py \
--config-path "${workspace}/conf" \ --config-path "${workspace}/conf" \
--config-name "${config}" \ --config-name "${config}" \

1
funasr/models/transformer/model.py
View File

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