fix typo
This commit is contained in:
parent
f453242f45
commit
23051041ce
|
|
@ -86,7 +86,7 @@ class WaveUnet(Model):
|
|||
num_channels = out_channels
|
||||
out_channels += initial_output_channels
|
||||
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:
|
||||
decoder = WavenetDecoder(num_channels+out_channels,num_channels)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue