Skip to content

Commit 03da871

Browse files
jimchen90Ji Chen
andauthored
MelResNet spacing and typing adjustment (#725)
* spacing and typing adjustment * remove import list Co-authored-by: Ji Chen <[email protected]>
1 parent 4318fc5 commit 03da871

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

torchaudio/models/_wavernn.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
from typing import Optional
2-
31
from torch import Tensor
42
from torch import nn
53

@@ -65,7 +63,7 @@ class _MelResNet(nn.Module):
6563
6664
Examples::
6765
>>> melresnet = _MelResNet(res_blocks=10, input_dims=100,
68-
hidden_dims=128, output_dims=128, pad=2)
66+
hidden_dims=128, output_dims=128, pad=2)
6967
>>> input = torch.rand(10, 100, 512)
7068
>>> output = melresnet(input)
7169
"""
@@ -99,9 +97,9 @@ def forward(self, x: Tensor) -> Tensor:
9997
10098
Shape:
10199
- x: :math:`(N, S, T)`.
102-
- output: :math:`(N, P, T-2*pad)`.
100+
- output: :math:`(N, P, T - 2 * pad)`.
103101
where N is the batch size, S is the number of input sequence,
104-
P is the number of ouput sequence, T is the length of input sequence.
102+
P is the number of output sequence, T is the length of input sequence.
105103
"""
106104

107105
return self.melresnet_model(x)

0 commit comments

Comments
 (0)