|
|
@ -185,6 +185,9 @@ class SqueezeWave(torch.nn.Module): |
|
|
|
n_early_size, WN_config): |
|
|
|
super(SqueezeWave, self).__init__() |
|
|
|
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_audio_channel = n_audio_channel |
|
|
|
self.n_early_every = n_early_every |
|
|
|