Browse Source

Update glow.py to fix upsample missing for denoiser

master
Daniel Muckerman 4 years ago
parent
commit
839ea13617
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      SqueezeWave/glow.py

+ 3
- 0
SqueezeWave/glow.py View File

@ -185,6 +185,9 @@ class SqueezeWave(torch.nn.Module):
n_early_size, WN_config): n_early_size, WN_config):
super(SqueezeWave, self).__init__() super(SqueezeWave, self).__init__()
assert(n_audio_channel % 2 == 0) assert(n_audio_channel % 2 == 0)
self.upsample = torch.nn.ConvTranspose1d(n_mel_channels,
n_mel_channels,
1024, stride=256)
self.n_flows = n_flows self.n_flows = n_flows
self.n_audio_channel = n_audio_channel self.n_audio_channel = n_audio_channel
self.n_early_every = n_early_every self.n_early_every = n_early_every

Loading…
Cancel
Save