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

minor changes to the scripts

Browse Source
This commit is contained in:
Quandwang 2022-07-21 15:58:09 +08:00
parent 1ad9862b39
commit 40eb8c43c9
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/conformer_ctc2/attention.py
View File

@ -62,6 +62,7 @@ class MultiheadAttention(nn.Module):
embed_dim, embed_dim,
num_heads, num_heads,
dropout=0.0, dropout=0.0,
bias=True,
add_bias_kv=False, add_bias_kv=False,
add_zero_attn=False, add_zero_attn=False,
kdim=None, kdim=None,
@ -78,8 +79,6 @@ class MultiheadAttention(nn.Module):
self._qkv_same_embed_dim = ( self._qkv_same_embed_dim = (
self.kdim == embed_dim and self.vdim == embed_dim self.kdim == embed_dim and self.vdim == embed_dim
) )
self.bias = True
bias = self.bias
self.num_heads = num_heads self.num_heads = num_heads
self.dropout = dropout self.dropout = dropout

2
egs/librispeech/ASR/conformer_ctc2/decode.py
View File

@ -125,7 +125,7 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--use-averaged-model", "--use-averaged-model",
type=str2bool, type=str2bool,
default=False, default=True,
help="Whether to load averaged model. Currently it only supports " help="Whether to load averaged model. Currently it only supports "
"using --epoch. If True, it would decode with the averaged model " "using --epoch. If True, it would decode with the averaged model "
"over the epoch range from `epoch-avg` (excluded) to `epoch`." "over the epoch range from `epoch-avg` (excluded) to `epoch`."