mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-10 13:31:13 +09:00
parent 45ce951b8c
commit de46c7b882
2 changed files with 30 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
View File

Binary file not shown.

38
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
View File

@ -105,18 +105,40 @@ def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None:
def add_rep_arguments(parser: argparse.ArgumentParser):
parser.add_argument(
"--num-encoder-layers",
type=str,
default="2,4,3,2,4",
help="Number of zipformer encoder layers, comma separated.",
"--decode-interval",
type=int,
default=200,
help="decode interval",
)
parser.add_argument(
"--feedforward-dims",
type=str,
default="1024,1024,2048,2048,1024",
help="Feedforward dimension of the zipformer encoder layers, comma separated.",
"--encoder-dim",
type=int,
default=768,
help="encoder embedding dimension",
)
parser.add_argument(
"--peak-enc-lr",
type=float,
default=0.0001,
help="The initial learning rate. This value should not need to be changed.",
)
parser.add_argument(
"--peak-dec-lr",
type=float,
default=0.001,
help="The initial learning rate. This value should not need to be changed.",
)
parser.add_argument(
"--multi-optim",
type=bool,
default=False,
help="use sperate optimizer (enc / dec)",
)
def add_model_arguments(parser: argparse.ArgumentParser):
parser.add_argument(