|
9 | 9 | class ResBlock(nn.Module): |
10 | 10 | r"""This is a ResNet block layer. This layer is based on the paper "Deep Residual Learning |
11 | 11 | 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. |
13 | 15 |
|
14 | 16 | Args: |
15 | 17 | num_dims: the number of compute dimensions in the input (default=128). |
@@ -49,8 +51,10 @@ def forward(self, x: Tensor) -> Tensor: |
49 | 51 |
|
50 | 52 |
|
51 | 53 | 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. |
54 | 58 |
|
55 | 59 | Args: |
56 | 60 | res_blocks: the number of ResBlock in stack (default=10). |
|
0 commit comments