From 23051041ce0304eda1acaa0e4498dd011c7f80e8 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 26 Sep 2022 12:24:53 +0530 Subject: [PATCH] fix typo --- enhancer/models/waveunet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhancer/models/waveunet.py b/enhancer/models/waveunet.py index 4a0e20f..62384fc 100644 --- a/enhancer/models/waveunet.py +++ b/enhancer/models/waveunet.py @@ -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)