update

examples/spectrum_dfnet_aishell/run.sh

@@ -8,7 +8,7 @@ sh run.sh --stage 2 --stop_stage 2 --system_version windows --file_folder_name f
 --speech_dir "E:/programmer/asr_datasets/aishell/data_aishell/wav/train"
 
 
-sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir \
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 

toolbox/torchaudio/models/spectrum_dfnet/modeling_spectrum_dfnet.py

@@ -845,6 +845,10 @@ class SpectrumDfNet(nn.Module):
         # spec_e shape: [batch_size, 1, time_steps, spec_bins, 2]
 
         spec_e[..., self.df_decoder.df_bins:, :] = spec_m[..., self.df_decoder.df_bins:, :]
+
+        spec_e = torch.squeeze(spec_e, dim=1)
+        spec_e = spec_e.permute(0, 2, 1, 3)
+        # spec_e shape: [batch_size, spec_bins, time_steps, 2]
         return spec_e, mask, lsnr