From 8f50a4cc490bb4edf2b567e59f498895cee50770 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期一, 24 四月 2023 19:40:32 +0800
Subject: [PATCH] update
---
egs/aishell/paraformer/conf/train_asr_paraformer_conformer_12e_6d_2048_256.yaml | 2 --
funasr/build_utils/build_args.py | 2 +-
egs/aishell/paraformer/run.sh | 1 +
3 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/egs/aishell/paraformer/conf/train_asr_paraformer_conformer_12e_6d_2048_256.yaml b/egs/aishell/paraformer/conf/train_asr_paraformer_conformer_12e_6d_2048_256.yaml
index 34c6fb3..24b2620 100644
--- a/egs/aishell/paraformer/conf/train_asr_paraformer_conformer_12e_6d_2048_256.yaml
+++ b/egs/aishell/paraformer/conf/train_asr_paraformer_conformer_12e_6d_2048_256.yaml
@@ -1,6 +1,4 @@
# network architecture
-# task related
-task_name: asr
# encoder related
encoder: conformer
encoder_conf:
diff --git a/egs/aishell/paraformer/run.sh b/egs/aishell/paraformer/run.sh
index ed6ddfe..b6ef733 100755
--- a/egs/aishell/paraformer/run.sh
+++ b/egs/aishell/paraformer/run.sh
@@ -162,6 +162,7 @@
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 \
diff --git a/funasr/build_utils/build_args.py b/funasr/build_utils/build_args.py
index 7c86acd..b19030d 100644
--- a/funasr/build_utils/build_args.py
+++ b/funasr/build_utils/build_args.py
@@ -88,5 +88,5 @@
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
--
Gitblit v1.9.1