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 2 --stop_stage 2 --system_version centos --file_folder_name file_dir \
+sh run.sh --stage 1 --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

@@ -221,7 +221,7 @@ def main():
     # training loop
 
     # state
-    average_pesq_metric = 10000000000
+    average_pesq_score = 10000000000
     average_loss = 10000000000
     average_ae_loss = 10000000000
     average_sc_loss = 10000000000
@@ -369,11 +369,11 @@ def main():
         # save metric
         if best_metric is None:
             best_idx_epoch = idx_epoch
-            best_metric = average_pesq_metric
-        elif average_pesq_metric > best_metric:
+            best_metric = average_pesq_score
+        elif average_pesq_score > best_metric:
             # great is better.
             best_idx_epoch = idx_epoch
-            best_metric = average_pesq_metric
+            best_metric = average_pesq_score
         else:
             pass
 
@@ -381,7 +381,7 @@ def main():
             "idx_epoch": idx_epoch,
             "best_idx_epoch": best_idx_epoch,
 
-            "pesq_metric": average_pesq_metric,
+            "pesq_score": average_pesq_score,
             "loss": average_loss,
             "ae_loss": average_ae_loss,
             "sc_loss": average_sc_loss,

toolbox/torchaudio/models/tcnn/__init__.py

@@ -0,0 +1,6 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+
+
+if __name__ == '__main__':
+    pass

toolbox/torchaudio/models/tcnn/modeling_tcnn.py

@@ -0,0 +1,15 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+"""
+https://github.com/LXP-Never/TCNN
+
+https://ieeexplore.ieee.org/abstract/document/8683634
+
+参考来源：
+https://github.com/WenzheLiu-Speech/awesome-speech-enhancement
+
+"""
+
+
+if __name__ == '__main__':
+    pass