deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

output variance

Browse Source
This commit is contained in:
James Betker 2022-05-02 00:10:33 -06:00
parent 3b074aac34
commit ab219fbefb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/audio/music/diffwave.py
View File

@ -144,7 +144,7 @@ class DiffWave(nn.Module):
for i in range(residual_layers)
])
self.skip_projection = Conv1d(residual_channels, residual_channels, 1)
self.output_projection = Conv1d(residual_channels, 1, 1)
self.output_projection = Conv1d(residual_channels, 2, 1)
nn.init.zeros_(self.output_projection.weight)
def forward(self, x, timesteps, spectrogram=None):
@ -176,4 +176,4 @@ def register_diffwave(opt_net, opt):
if __name__ == '__main__':
model = DiffWave()
model(torch.randn(2,1,20000), torch.tensor([500,3999]), torch.randn(2,128,78))
model(torch.randn(2,1,65536), torch.tensor([500,3999]), torch.randn(2,128,256))