Revert "Update decoder.py"

This reverts commit 0c8b3975170f74b61e98885771eccd7f94679608.
This commit is contained in:
jinzr 2023-10-23 21:38:33 +08:00
parent 0c8b397517
commit 8fbd79e3c0

View File

@ -95,10 +95,6 @@ class Decoder(nn.Module):
max_abs=1.0,
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:
"""