From 7698c3638aab92466b8855e2f19a79b4f9d969ab Mon Sep 17 00:00:00 2001 From: Ji Chen Date: Wed, 17 Jun 2020 09:39:39 -0700 Subject: [PATCH 1/2] spacing and typing adjustment --- torchaudio/models/_wavernn.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/torchaudio/models/_wavernn.py b/torchaudio/models/_wavernn.py index a08af175f8..4fedea3384 100644 --- a/torchaudio/models/_wavernn.py +++ b/torchaudio/models/_wavernn.py @@ -1,4 +1,4 @@ -from typing import Optional +from typing import List from torch import Tensor from torch import nn @@ -65,7 +65,7 @@ class _MelResNet(nn.Module): Examples:: >>> melresnet = _MelResNet(res_blocks=10, input_dims=100, - hidden_dims=128, output_dims=128, pad=2) + hidden_dims=128, output_dims=128, pad=2) >>> input = torch.rand(10, 100, 512) >>> output = melresnet(input) """ @@ -99,9 +99,9 @@ def forward(self, x: Tensor) -> Tensor: Shape: - x: :math:`(N, S, T)`. - - output: :math:`(N, P, T-2*pad)`. + - output: :math:`(N, P, T - 2 * pad)`. where N is the batch size, S is the number of input sequence, - P is the number of ouput sequence, T is the length of input sequence. + P is the number of output sequence, T is the length of input sequence. """ return self.melresnet_model(x) From de25042efbc986721ac8826cf158f02bbecce569 Mon Sep 17 00:00:00 2001 From: Ji Chen Date: Wed, 17 Jun 2020 11:19:23 -0700 Subject: [PATCH 2/2] remove import list --- torchaudio/models/_wavernn.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/torchaudio/models/_wavernn.py b/torchaudio/models/_wavernn.py index 4fedea3384..04155fb87c 100644 --- a/torchaudio/models/_wavernn.py +++ b/torchaudio/models/_wavernn.py @@ -1,5 +1,3 @@ -from typing import List - from torch import Tensor from torch import nn