This commit is contained in:
嘉渊 2023-04-24 19:40:32 +08:00
parent 9847fd24a8
commit 8f50a4cc49
3 changed files with 2 additions and 3 deletions

View File

@ -1,6 +1,4 @@
# network architecture
# task related
task_name: asr
# encoder related
encoder: conformer
encoder_conf:

View File

@ -162,6 +162,7 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
local_rank=$i
gpu_id=$(echo $CUDA_VISIBLE_DEVICES | cut -d',' -f$[$i+1])
train.py \
--task_name asr \
--gpu_id $gpu_id \
--use_preprocessor true \
--token_type char \

View File

@ -88,5 +88,5 @@ def build_args(args, parser, extra_task_params):
if not any(action.dest == a.dest for a in task_parser._actions):
task_parser._add_action(action)
task_args = parser.parse_args(extra_task_params)
task_args = task_parser.parse_args(extra_task_params)
return task_args