Skip to content

Commit 490081c

Browse files
author
Ji Chen
committed
format update
1 parent a8528bf commit 490081c

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

torchaudio/models/_wavernn.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99

1010
class _ResBlock(nn.Module):
11-
r"""ResNet block layer based on
11+
r"""ResNet block layer based on
1212
`"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_
1313
1414
Args:
@@ -170,7 +170,11 @@ def __init__(self,
170170
up_layers = []
171171
for scale in upsample_scales:
172172
stretch = _Stretch2d(scale, 1)
173-
conv = nn.Conv2d(in_channels=1, out_channels=1, kernel_size=(1, scale * 2 + 1), padding=(0, scale), bias=False)
173+
conv = nn.Conv2d(in_channels=1,
174+
out_channels=1,
175+
kernel_size=(1, scale * 2 + 1),
176+
padding=(0, scale),
177+
bias=False)
174178
conv.weight.data.fill_(1. / (scale * 2 + 1))
175179
up_layers.append(stretch)
176180
up_layers.append(conv)

0 commit comments

Comments
 (0)