From b7e34131d22a6f69325facf03e26701415be01a4 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期一, 24 四月 2023 17:21:22 +0800
Subject: [PATCH] update
---
funasr/bin/train.py | 7 +++----
1 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index f056b08..474e857 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -19,7 +19,6 @@
from funasr.torch_utils.model_summary import model_summary
from funasr.torch_utils.pytorch_version import pytorch_cudnn_version
from funasr.torch_utils.set_all_random_seed import set_all_random_seed
-from funasr.utils import config_argparse
from funasr.utils.prepare_data import prepare_data
from funasr.utils.types import str2bool
from funasr.utils.types import str_or_none
@@ -27,7 +26,7 @@
def get_parser():
- parser = config_argparse.ArgumentParser(
+ parser = argparse.ArgumentParser(
description="FunASR Common Training Parser",
)
@@ -390,8 +389,8 @@
parser = get_parser()
args, extra_task_params = parser.parse_known_args()
if extra_task_params:
- task_args = build_args(args, extra_task_params)
- args = argparse.Namespace(**vars(args), **vars(task_args))
+ args = build_args(args, parser, extra_task_params)
+ # args = argparse.Namespace(**vars(args), **vars(task_args))
# set random seed
set_all_random_seed(args.seed)
--
Gitblit v1.9.1