Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on 7 days ago

Commit

8f4f9ae

1 Parent(s): 6a8bc1c

update

Files changed (3) hide show

examples/nx_clean_unet/yaml/config.yaml CHANGED Viewed

@@ -12,8 +12,8 @@ down_sampling_hidden_channels: 64
 down_sampling_kernel_size: 4
 down_sampling_stride: 2
-causal_in_channels: 64
-causal_out_channels: 64
 causal_kernel_size: 3
 causal_bias: false
 causal_separable: true

 down_sampling_kernel_size: 4
 down_sampling_stride: 2
+causal_in_channels: 1
+causal_out_channels: 1
 causal_kernel_size: 3
 causal_bias: false
 causal_separable: true

toolbox/torchaudio/models/nx_clean_unet/causal_convolution/causal_conv2d.py CHANGED Viewed

@@ -236,7 +236,7 @@ def main2():
 def main():
     causal = CausalConv2dEncoder(
         in_channels=1,
-        out_channels=64,
         kernel_size=3,
         bias=False,
         separable=True,

 def main():
     causal = CausalConv2dEncoder(
         in_channels=1,
+        out_channels=1,
         kernel_size=3,
         bias=False,
         separable=True,

toolbox/torchaudio/models/nx_clean_unet/yaml/config.yaml CHANGED Viewed

@@ -21,8 +21,8 @@ down_sampling_hidden_channels: 64
 down_sampling_kernel_size: 4
 down_sampling_stride: 2
-causal_in_channels: 64
-causal_out_channels: 64
 causal_kernel_size: 3
 causal_bias: false
 causal_separable: true

 down_sampling_kernel_size: 4
 down_sampling_stride: 2
+causal_in_channels: 1
+causal_out_channels: 1
 causal_kernel_size: 3
 causal_bias: false
 causal_separable: true