mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-06 23:54:17 +00:00
Revert "Update decoder.py"
This reverts commit 0c8b3975170f74b61e98885771eccd7f94679608.
This commit is contained in:
parent
0c8b397517
commit
8fbd79e3c0
@ -95,10 +95,6 @@ class Decoder(nn.Module):
|
|||||||
max_abs=1.0,
|
max_abs=1.0,
|
||||||
prob=0.05,
|
prob=0.05,
|
||||||
)
|
)
|
||||||
else:
|
|
||||||
# To avoid `RuntimeError: Module 'Decoder' has no attribute 'conv'`
|
|
||||||
# when inference with torch.jit.script and context_size == 1
|
|
||||||
self.conv = nn.Identity()
|
|
||||||
|
|
||||||
def forward(self, y: torch.Tensor, need_pad: bool = True) -> torch.Tensor:
|
def forward(self, y: torch.Tensor, need_pad: bool = True) -> torch.Tensor:
|
||||||
"""
|
"""
|
||||||
|
Loading…
x
Reference in New Issue
Block a user