update

examples/nx_mpnet/run.sh

@@ -3,7 +3,7 @@
 : <<'END'
 
 
-sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name nx-mpnet-aishell-20250224 \
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name nx-mpnet-aishell \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train" \
 --max_epochs 100 \

examples/nx_mpnet/yaml/config.yaml

@@ -22,8 +22,8 @@ tsfm_dropout_rate: 0.0
 tsfm_max_time_relative_position: 2048
 tsfm_max_freq_relative_position: 256
 tsfm_chunk_size: 1
-tsfm_num_left_chunks: 96
-tsfm_num_right_chunks: 3
+tsfm_num_left_chunks: 64
+tsfm_num_right_chunks: 32
 
 discriminator_dim: 32
 discriminator_in_channel: 2