HoneyTian commited on
Commit
44256c5
·
1 Parent(s): 294430e
Files changed (1) hide show
  1. examples/nx_clean_unet/run.sh +4 -1
examples/nx_clean_unet/run.sh CHANGED
@@ -20,7 +20,7 @@ sh run.sh --stage 5 --stop_stage 5 --system_version centos --file_folder_name fi
20
  sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-nx-speech-20250224 \
21
  --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
22
  --speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech" \
23
- --max_epochs 1
24
 
25
 
26
  END
@@ -41,6 +41,8 @@ limit=10
41
  noise_dir=/data/tianxing/HuggingDatasets/nx_noise/data/noise
42
  speech_dir=/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train
43
 
 
 
44
  nohup_name=nohup.out
45
 
46
  # model params
@@ -107,6 +109,7 @@ if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
107
  --speech_dir "${speech_dir}" \
108
  --train_dataset "${train_dataset}" \
109
  --valid_dataset "${valid_dataset}" \
 
110
 
111
  fi
112
 
 
20
  sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-nx-speech-20250224 \
21
  --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
22
  --speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech" \
23
+ --max_epochs 100 --max_count 10000
24
 
25
 
26
  END
 
41
  noise_dir=/data/tianxing/HuggingDatasets/nx_noise/data/noise
42
  speech_dir=/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train
43
 
44
+ max_count=10000000
45
+
46
  nohup_name=nohup.out
47
 
48
  # model params
 
109
  --speech_dir "${speech_dir}" \
110
  --train_dataset "${train_dataset}" \
111
  --valid_dataset "${valid_dataset}" \
112
+ --max_count "${max_count}" \
113
 
114
  fi
115