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

add tanh_on_mem option

Browse Source
This commit is contained in:
yaozengwei 2022-05-10 11:48:22 +08:00
parent e3a29b17f3
commit 5de9d0a19a
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -130,6 +130,13 @@ def add_model_arguments(parser: argparse.ArgumentParser):
help="Number of entries in the memory for the Emformer", help="Number of entries in the memory for the Emformer",
) )
parser.add_argument(
"--tanh-on-mem",
type=str2bool,
default=False,
help="Whether to apply tanh on memory",
)
def get_parser(): def get_parser():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
@ -369,6 +376,7 @@ def get_encoder_model(params: AttributeDict) -> nn.Module:
chunk_length=params.chunk_length, chunk_length=params.chunk_length,
right_context_length=params.right_context_length, right_context_length=params.right_context_length,
max_memory_size=params.memory_size, max_memory_size=params.memory_size,
tanh_on_mem=params.tanh_on_mem,
) )
return encoder return encoder