mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 10:02:22 +00:00
Increase scale on Scale from 4 to 20
This commit is contained in:
parent
503f8d521c
commit
3207bd98a9
@ -48,12 +48,12 @@ class Conv2dSubsampling(nn.Module):
|
||||
in_channels=1, out_channels=odim, kernel_size=3, stride=2
|
||||
),
|
||||
nn.ReLU(),
|
||||
ExpScale(odim, 1, 1, speed=4.0),
|
||||
ExpScale(odim, 1, 1, speed=20.0),
|
||||
nn.Conv2d(
|
||||
in_channels=odim, out_channels=odim, kernel_size=3, stride=2
|
||||
),
|
||||
nn.ReLU(),
|
||||
ExpScale(odim, 1, 1, speed=4.0),
|
||||
ExpScale(odim, 1, 1, speed=20.0),
|
||||
)
|
||||
self.out = nn.Linear(odim * (((idim - 1) // 2 - 1) // 2), odim)
|
||||
self.out_norm = nn.LayerNorm(odim, elementwise_affine=False)
|
||||
|
@ -156,7 +156,7 @@ class ConformerEncoderLayer(nn.Module):
|
||||
|
||||
self.feed_forward = nn.Sequential(
|
||||
nn.Linear(d_model, dim_feedforward),
|
||||
ExpScaleSwish(dim_feedforward, speed=4.0),
|
||||
ExpScaleSwish(dim_feedforward, speed=20.0),
|
||||
nn.Dropout(dropout),
|
||||
nn.Linear(dim_feedforward, d_model),
|
||||
)
|
||||
@ -164,7 +164,7 @@ class ConformerEncoderLayer(nn.Module):
|
||||
self.feed_forward_macaron = nn.Sequential(
|
||||
nn.Linear(d_model, dim_feedforward),
|
||||
Swish(),
|
||||
ExpScaleSwish(dim_feedforward, speed=4.0),
|
||||
ExpScaleSwish(dim_feedforward, speed=20.0),
|
||||
nn.Dropout(dropout),
|
||||
nn.Linear(dim_feedforward, d_model),
|
||||
)
|
||||
|
@ -110,7 +110,7 @@ def get_parser():
|
||||
parser.add_argument(
|
||||
"--exp-dir",
|
||||
type=str,
|
||||
default="transducer_stateless/specaugmod_baseline_randcombine1_expscale2",
|
||||
default="transducer_stateless/specaugmod_baseline_randcombine1_expscale3",
|
||||
help="""The experiment dir.
|
||||
It specifies the directory where all training related
|
||||
files, e.g., checkpoints, log, etc, are saved
|
||||
|
Loading…
x
Reference in New Issue
Block a user