update

examples/clean_unet_aishell/run.sh

@@ -12,7 +12,7 @@ sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name fi
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 
-sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir \
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 

examples/clean_unet_aishell/step_2_train_model.py

@@ -233,7 +233,7 @@ def main():
     patience_count = 0
 
     logger.info("training")
-    for idx_epoch in range(args.max_epochs):
+    for idx_epoch in range(max(0, last_epoch), args.max_epochs):
         # train
         model.train()