Skip to content

GitLab

Unverified Commit 937d52f8 authored by jimchen90's avatar jimchen90 Committed by GitHub
Browse files

Fix output type of upsampling (#801) 



Fix output type of upsampling
Co-authored-by: default avatarJi Chen <jimchen90@devfair0160.h2.fair>
parent e82cc350
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchaudio/models/_wavernn.py
View file @ 937d52f8
from typing import List from typing import List, Tuple
import torch import torch
from torch import Tensor from torch import Tensor
...@@ -172,7 +172,7 @@ class _UpsampleNetwork(nn.Module): ...@@ -172,7 +172,7 @@ class _UpsampleNetwork(nn.Module):
up_layers.append(conv) up_layers.append(conv)
self.upsample_layers = nn.Sequential(*up_layers) self.upsample_layers = nn.Sequential(*up_layers)
def forward(self, specgram: Tensor) -> Tensor: def forward(self, specgram: Tensor) -> Tuple[Tensor, Tensor]:
r"""Pass the input through the _UpsampleNetwork layer. r"""Pass the input through the _UpsampleNetwork layer.
Args: Args:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment