mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
update
This commit is contained in:
parent
362a6de3c8
commit
6c8e839742
@ -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}" \
|
||||
|
||||
@ -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:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user