fix typo
This commit is contained in:
parent
f453242f45
commit
23051041ce
|
|
@ -86,7 +86,7 @@ class WaveUnet(Model):
|
||||||
num_channels = out_channels
|
num_channels = out_channels
|
||||||
out_channels += initial_output_channels
|
out_channels += initial_output_channels
|
||||||
if layer == num_layers -1 :
|
if layer == num_layers -1 :
|
||||||
decoder = WavenetDecoder(num_layers * initial_output_channels + num_channels,inp_channels)
|
decoder = WavenetDecoder(num_layers * initial_output_channels + num_channels,num_channels)
|
||||||
else:
|
else:
|
||||||
decoder = WavenetDecoder(num_channels+out_channels,num_channels)
|
decoder = WavenetDecoder(num_channels+out_channels,num_channels)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue