mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-07 08:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
jinzr 2023-10-11 09:55:05 +08:00
parent b7bcece98d
commit 7c37a8f933
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
egs/aishell/ASR/pruned_transducer_stateless7_streaming/train.py
View File

@ -257,6 +257,16 @@ def get_parser():
""", """,
) )
parser.add_argument(
"--lang-dir",
type=str,
default="data/lang_char",
help="""The lang dir
It contains language related input files such as
"lexicon.txt"
""",
)
parser.add_argument( parser.add_argument(
"--base-lr", type=float, default=0.05, help="The base learning rate." "--base-lr", type=float, default=0.05, help="The base learning rate."
) )
@ -1224,6 +1234,7 @@ def main():
AishellAsrDataModule.add_arguments(parser) AishellAsrDataModule.add_arguments(parser)
args = parser.parse_args() args = parser.parse_args()
args.exp_dir = Path(args.exp_dir) args.exp_dir = Path(args.exp_dir)
args.lang_dir = Path(args.lang_dir)
world_size = args.world_size world_size = args.world_size
assert world_size >= 1 assert world_size >= 1