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

View File

@ -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}" \

View File

@ -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: