mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set scaling on SwishExpScale

Browse Source
This commit is contained in:
Daniel Povey 2022-03-11 20:12:45 +08:00
parent cc558faf26
commit 2d3a76292d
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/conformer_ctc/subsampling.py
View File

@ -255,7 +255,9 @@ class SwishExpScale(torch.nn.Module):
def __init__(self, *shape, speed: float = 1.0, in_scale: float = 1.0): def __init__(self, *shape, speed: float = 1.0, in_scale: float = 1.0):
super(SwishExpScale, self).__init__() super(SwishExpScale, self).__init__()
self.in_scale = in_scale self.in_scale = in_scale
self.scale = nn.Parameter(torch.zeros(*shape)) initial_log_scale = torch.tensor(1.0 / in_scale).log() / speed
initial_log_scale = (torch.ones(*shape) * initial_log_scale).detach()
self.scale = nn.Parameter(initial_log_scale)
self.speed = speed self.speed = speed
def forward(self, x: Tensor) -> Tensor: def forward(self, x: Tensor) -> Tensor: