| | |
|---|
| | | valid_set=dev |
|---|
| | | test_sets="dev test" |
|---|
| | | |
|---|
| | | config=paraformer_conformer_12e_6d_2048_256.yaml |
|---|
| | | config=transformer_12e_6d_2048_256.yaml |
|---|
| | | model_dir="baseline_$(basename "${config}" .yaml)_${lang}_${token_type}_${tag}" |
|---|
| | | |
|---|
| | | |
|---|
| | |
|---|
| | | echo "stage 4: ASR Training" |
|---|
| | | |
|---|
| | | mkdir -p ${exp_dir}/exp/${model_dir} |
|---|
| | | log_file="${exp_dir}/exp/${model_dir}/train.log.txt" |
|---|
| | | current_time=$(date "+%Y-%m-%d_%H-%M") |
|---|
| | | log_file="${exp_dir}/exp/${model_dir}/train.log.txt.${current_time}" |
|---|
| | | 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 \ |
|---|
| | | --nnodes 1 \ |
|---|
| | | --nproc_per_node ${gpu_num} \ |
|---|
