Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on Feb 28

Commit

f91bf4a

1 Parent(s): 23dc849

update

Files changed (1) hide show

toolbox/torchaudio/models/nx_clean_unet/modeling_nx_clean_unet.py CHANGED Viewed

@@ -58,7 +58,7 @@ class DownSampling(nn.Module):
                 kernel_size=kernel_size,
                 stride=stride,
             )
-            self.down_sampling_block_list.append(down_sampling_block)
             in_channels = hidden_channels
         self.down_sampling_block_list = nn.ModuleList(modules=down_sampling_block_list)
@@ -121,7 +121,7 @@ class UpSampling(nn.Module):
                 stride=stride,
                 do_relu=True,
             )
-            self.up_sampling_block_list.append(up_sampling_block)
         else:
             up_sampling_block = UpSamplingBlock(
                 out_channels=out_channels,
@@ -130,7 +130,7 @@ class UpSampling(nn.Module):
                 stride=stride,
                 do_relu=False,
             )
-            self.up_sampling_block_list.append(up_sampling_block)
         self.up_sampling_block_list = nn.ModuleList(modules=up_sampling_block_list)

                 kernel_size=kernel_size,
                 stride=stride,
             )
+            down_sampling_block_list.append(down_sampling_block)
             in_channels = hidden_channels
         self.down_sampling_block_list = nn.ModuleList(modules=down_sampling_block_list)
                 stride=stride,
                 do_relu=True,
             )
+            up_sampling_block_list.append(up_sampling_block)
         else:
             up_sampling_block = UpSamplingBlock(
                 out_channels=out_channels,
                 stride=stride,
                 do_relu=False,
             )
+            up_sampling_block_list.append(up_sampling_block)
         self.up_sampling_block_list = nn.ModuleList(modules=up_sampling_block_list)