Spaces:
Running
Running
add microphone audio input
Browse files
examples/dfnet/run.sh
CHANGED
@@ -6,8 +6,8 @@ sh run.sh --stage 2 --stop_stage 2 --system_version windows --file_folder_name f
|
|
6 |
--noise_dir "E:/Users/tianx/HuggingDatasets/nx_noise/data/noise" \
|
7 |
--speech_dir "E:/Users/tianx/HuggingDatasets/nx_noise/data/speech"
|
8 |
|
9 |
-
sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name dfnet-dns3 \
|
10 |
-
--noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise
|
11 |
--speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech/dns3-speech"
|
12 |
|
13 |
|
|
|
6 |
--noise_dir "E:/Users/tianx/HuggingDatasets/nx_noise/data/noise" \
|
7 |
--speech_dir "E:/Users/tianx/HuggingDatasets/nx_noise/data/speech"
|
8 |
|
9 |
+
sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name dfnet-nx-dns3 \
|
10 |
+
--noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
|
11 |
--speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech/dns3-speech"
|
12 |
|
13 |
|
toolbox/torchaudio/modules/freq_bands/erb_bands.py
CHANGED
@@ -122,8 +122,10 @@ class ErbBands(nn.Module):
|
|
122 |
self.min_freq_bins_for_erb = min_freq_bins_for_erb
|
123 |
|
124 |
erb_fb, erb_fb_inv = self.init_erb_fb()
|
125 |
-
|
126 |
-
|
|
|
|
|
127 |
|
128 |
def init_erb_fb(self):
|
129 |
erb_widths = ErbBandsNumpy.get_erb_widths(
|
|
|
122 |
self.min_freq_bins_for_erb = min_freq_bins_for_erb
|
123 |
|
124 |
erb_fb, erb_fb_inv = self.init_erb_fb()
|
125 |
+
erb_fb = torch.tensor(erb_fb, dtype=torch.float32, requires_grad=False)
|
126 |
+
erb_fb_inv = torch.tensor(erb_fb_inv, dtype=torch.float32, requires_grad=False)
|
127 |
+
self.erb_fb = nn.Parameter(erb_fb, requires_grad=False)
|
128 |
+
self.erb_fb_inv = nn.Parameter(erb_fb_inv, requires_grad=False)
|
129 |
|
130 |
def init_erb_fb(self):
|
131 |
erb_widths = ErbBandsNumpy.get_erb_widths(
|