mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-19 05:54:20 +00:00
minor changes to the scripts
This commit is contained in:
parent
1ad9862b39
commit
40eb8c43c9
@ -62,6 +62,7 @@ class MultiheadAttention(nn.Module):
|
||||
embed_dim,
|
||||
num_heads,
|
||||
dropout=0.0,
|
||||
bias=True,
|
||||
add_bias_kv=False,
|
||||
add_zero_attn=False,
|
||||
kdim=None,
|
||||
@ -78,8 +79,6 @@ class MultiheadAttention(nn.Module):
|
||||
self._qkv_same_embed_dim = (
|
||||
self.kdim == embed_dim and self.vdim == embed_dim
|
||||
)
|
||||
self.bias = True
|
||||
bias = self.bias
|
||||
|
||||
self.num_heads = num_heads
|
||||
self.dropout = dropout
|
||||
|
@ -125,7 +125,7 @@ def get_parser():
|
||||
parser.add_argument(
|
||||
"--use-averaged-model",
|
||||
type=str2bool,
|
||||
default=False,
|
||||
default=True,
|
||||
help="Whether to load averaged model. Currently it only supports "
|
||||
"using --epoch. If True, it would decode with the averaged model "
|
||||
"over the epoch range from `epoch-avg` (excluded) to `epoch`."
|
||||
|
Loading…
x
Reference in New Issue
Block a user