HoneyTian commited on
Commit
a02b6ed
·
1 Parent(s): ac26d56

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/dns3-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
- self.erb_fb = torch.tensor(erb_fb, dtype=torch.float32, requires_grad=False)
126
- self.erb_fb_inv = torch.tensor(erb_fb_inv, dtype=torch.float32, requires_grad=False)
 
 
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(