update

examples/vm_sound_classification/run.sh

@@ -12,8 +12,8 @@ sh run.sh --stage 2 --stop_stage 2 --system_version windows --file_folder_name f
 E:/Users/tianx/HuggingDatasets/vm_sound_classification/data/wav_finished/id-ID/wav_finished/*/*.wav" \
 --label_plan 4
 
-sh run.sh --stage 0 --stop_stage 5 --system_version centos --file_folder_name file_dir --final_model_name vm_sound_classification3-ch16 \
---filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/*/wav_finished/*/*.wav" --label_plan 3
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name vm_sound_classification2-ch4 \
+--filename_patterns "/data/tianxing/PycharmProjects/datasets/voicemail/*/wav_finished/*/*.wav" --label_plan 2
 
 "
 

toolbox/torchaudio/models/cnn_audio_classifier/modeling_cnn_audio_classifier.py

@@ -346,7 +346,6 @@ class WaveClassifierPretrainedModel(WaveClassifier):
         super(WaveClassifierPretrainedModel, self).__init__(
             wave_encoder=WaveEncoder(
                 mel_spectrogram_param=config.mel_spectrogram_param,
-                spec_augment_param=config.spec_augment_param,
                 conv1d_block_param_list=config.conv1d_block_param_list,
                 conv2d_block_param_list=config.conv2d_block_param_list,
             ),