This commit is contained in:
shahules786 2022-09-26 12:24:53 +05:30
parent f453242f45
commit 23051041ce
1 changed files with 1 additions and 1 deletions

View File

@ -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)