Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on Feb 28

Commit

dc7c9fa

1 Parent(s): c37a8f9

update

Files changed (2) hide show

examples/nx_clean_unet/run.sh CHANGED Viewed

@@ -15,7 +15,7 @@ sh run.sh --stage 3 --stop_stage 3 --system_version centos --file_folder_name fi
 sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name nx-clean-unet-aishell-20250228 \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train" \
---max_epochs 100
 sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-nx-speech-20250224 \

 sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name nx-clean-unet-aishell-20250228 \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train" \
+--max_epochs 100 --max_count 10000
 sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-nx-speech-20250224 \

examples/nx_clean_unet/yaml/config.yaml CHANGED Viewed

@@ -13,7 +13,7 @@ down_sampling_kernel_size: 4
 down_sampling_stride: 2
 tsfm_hidden_size: 2048
-tsfm_attention_heads: 12
 tsfm_num_blocks: 6
 tsfm_dropout_rate: 0.1
 tsfm_max_length: 1024

 down_sampling_stride: 2
 tsfm_hidden_size: 2048
+tsfm_attention_heads: 16
 tsfm_num_blocks: 6
 tsfm_dropout_rate: 0.1
 tsfm_max_length: 1024