mirrors/icefall
Archived
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-10 13:36:01 +09:00
parent 0291ce4eff
commit a6b159f56f
2 changed files with 30 additions and 24 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.

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

@ -105,17 +105,23 @@ def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None:
def add_rep_arguments(parser: argparse.ArgumentParser): def add_rep_arguments(parser: argparse.ArgumentParser):
parser.add_argument( parser.add_argument(
"--decode-interval", "--wandb",
type=bool,
default=False,
help="Use wandb for MLOps",
)
parser.add_argument(
"--accum-grads",
type=int, type=int,
default=200, default=1,
help="decode interval", help="accum-grad num.",
) )
parser.add_argument( parser.add_argument(
"--encoder-dim", "--multi-optim",
type=int, type=bool,
default=768, default=False,
help="encoder embedding dimension", help="use sperate optimizer (enc / dec)",
) )
parser.add_argument( parser.add_argument(
@ -132,41 +138,41 @@ def add_rep_arguments(parser: argparse.ArgumentParser):
help="The initial learning rate. This value should not need to be changed.", 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)",
)
parser.add_argument(
"--accum-grads",
type=int,
default=1,
help="accum-grad num.",
)
parser.add_argument( parser.add_argument(
"--encoder-type", "--encoder-type",
type=str, type=str,
default='d2v', default='d2v',
help="Type of encoder (e.g. conformer, w2v, d2v...", help="Type of encoder (e.g. conformer, w2v, d2v...",
) )
parser.add_argument( parser.add_argument(
"--additional-block", "--encoder-dim",
type=bool, type=int,
default=False, default=768,
help="encoder embedding dimension",
) )
parser.add_argument( parser.add_argument(
"--freeze-finetune-updates", "--freeze-finetune-updates",
type=int, type=int,
default=0 default=0
) )
parser.add_argument( parser.add_argument(
"--wandb", "--additional-block",
type=bool, type=bool,
default=False, default=False,
help="Use wandb for MLOps",
) )
parser.add_argument(
"--decode-interval",
type=int,
default=200,
help="decode interval",
)
def add_model_arguments(parser: argparse.ArgumentParser): def add_model_arguments(parser: argparse.ArgumentParser):