speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
嘉渊 2023-04-24 19:50:07 +08:00
parent 420849cfb3
commit 6427c834df
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
funasr/bin/train.py
View File

@ -19,6 +19,7 @@ from funasr.text.phoneme_tokenizer import g2p_choices
from funasr.torch_utils.model_summary import model_summary from funasr.torch_utils.model_summary import model_summary
from funasr.torch_utils.pytorch_version import pytorch_cudnn_version 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.torch_utils.set_all_random_seed import set_all_random_seed
from funasr.utils.nested_dict_action import NestedDictAction
from funasr.utils.prepare_data import prepare_data from funasr.utils.prepare_data import prepare_data
from funasr.utils.types import str2bool from funasr.utils.types import str2bool
from funasr.utils.types import str_or_none from funasr.utils.types import str_or_none
@ -302,6 +303,32 @@ def get_parser():
help="Apply preprocessing to data or not", help="Apply preprocessing to data or not",
) )
# optimization related
parser.add_argument(
"--optim",
type=lambda x: x.lower(),
default="adam",
help="The optimizer type",
)
parser.add_argument(
"--optim_conf",
action=NestedDictAction,
default=dict(),
help="The keyword arguments for optimizer",
)
parser.add_argument(
"--scheduler",
type=lambda x: str_or_none(x.lower()),
default=None,
help="The lr scheduler type",
)
parser.add_argument(
"--scheduler_conf",
action=NestedDictAction,
default=dict(),
help="The keyword arguments for lr scheduler",
)
# most task related # most task related
parser.add_argument( parser.add_argument(
"--init", "--init",