Skip to content

Commit 79b715c

Browse files
author
Ji Chen
committed
update reference in docstrings
1 parent 13e631d commit 79b715c

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

torchaudio/models/wavernn.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,9 @@
99
class ResBlock(nn.Module):
1010
r"""This is a ResNet block layer. This layer is based on the paper "Deep Residual Learning
1111
for Image Recognition". Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun. CVPR, 2016.
12-
It is a block used in WaveRNN(https://github.com/G-Wang/WaveRNN-Pytorch).
12+
It is a block used in WaveRNN. WaveRNN is based on the paper "Efficient Neural Audio Synthesis".
13+
Nal Kalchbrenner, Erich Elsen, Karen Simonyan, Seb Noury, Norman Casagrande, Edward Lockhart,
14+
Florian Stimberg, Aaron van den Oord, Sander Dieleman, Koray Kavukcuoglu. arXiv:1802.08435, 2018.
1315
1416
Args:
1517
num_dims: the number of compute dimensions in the input (default=128).
@@ -49,8 +51,10 @@ def forward(self, x: Tensor) -> Tensor:
4951

5052

5153
class MelResNet(nn.Module):
52-
r"""This is a MelResNet layer based on a stack of ResBlocks. It is a block used in WaveRNN
53-
(https://github.com/G-Wang/WaveRNN-Pytorch).
54+
r"""This is a MelResNet layer based on a stack of ResBlocks. It is a block used in WaveRNN.
55+
WaveRNN is based on the paper "Efficient Neural Audio Synthesis". Nal Kalchbrenner, Erich Elsen,
56+
Karen Simonyan, Seb Noury, Norman Casagrande, Edward Lockhart, Florian Stimberg, Aaron van den Oord,
57+
Sander Dieleman, Koray Kavukcuoglu. arXiv:1802.08435, 2018.
5458
5559
Args:
5660
res_blocks: the number of ResBlock in stack (default=10).

0 commit comments

Comments
 (0)