HoneyTian commited on
Commit
de8138b
·
1 Parent(s): 8885485
examples/nx_clean_unet/run.sh CHANGED
@@ -15,7 +15,8 @@ sh run.sh --stage 3 --stop_stage 3 --system_version centos --file_folder_name fi
15
  sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir2 --final_model_name nx-clean-unet-aishell-20250228 \
16
  --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
17
  --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train" \
18
- --max_epochs 100 --max_count 10000
 
19
 
20
 
21
  sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-nx-speech-20250224 \
 
15
  sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir2 --final_model_name nx-clean-unet-aishell-20250228 \
16
  --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
17
  --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train" \
18
+ --max_epochs 100
19
+ --max_count 10000
20
 
21
 
22
  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
@@ -25,7 +25,7 @@ discriminator_in_channel: 2
25
 
26
  compress_factor: 0.3
27
 
28
- batch_size: 4
29
  learning_rate: 0.0005
30
  adam_b1: 0.8
31
  adam_b2: 0.99
 
25
 
26
  compress_factor: 0.3
27
 
28
+ batch_size: 16
29
  learning_rate: 0.0005
30
  adam_b1: 0.8
31
  adam_b2: 0.99