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

monir style fix

Browse Source
This commit is contained in:
yaozengwei 2022-05-28 19:17:19 +08:00
parent 26b443eac3
commit 603b707fa8
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/librispeech/ASR/conv_emformer_transducer_stateless/train.py
View File

@ -71,8 +71,8 @@ import torch
import torch.multiprocessing as mp import torch.multiprocessing as mp
import torch.nn as nn import torch.nn as nn
from asr_datamodule import LibriSpeechAsrDataModule from asr_datamodule import LibriSpeechAsrDataModule
from emformer import Emformer
from decoder import Decoder from decoder import Decoder
from emformer import Emformer
from joiner import Joiner from joiner import Joiner
from lhotse.cut import Cut from lhotse.cut import Cut
from lhotse.dataset.sampling.base import CutSampler from lhotse.dataset.sampling.base import CutSampler
@ -141,7 +141,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
type=int, type=int,
default=32, default=32,
help="""Number of frames before subsampling for left context help="""Number of frames before subsampling for left context
in the Emformer.""" in the Emformer.""",
) )
parser.add_argument( parser.add_argument(
@ -149,7 +149,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
type=int, type=int,
default=32, default=32,
help="""Number of frames before subsampling for each chunk help="""Number of frames before subsampling for each chunk
in the Emformer.""" in the Emformer.""",
) )
parser.add_argument( parser.add_argument(
@ -157,7 +157,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
type=int, type=int,
default=8, default=8,
help="""Number of frames before subsampling for right context help="""Number of frames before subsampling for right context
in the Emformer.""" in the Emformer.""",
) )
parser.add_argument( parser.add_argument(