Initial commit

README.md

@@ -3,9 +3,11 @@ tags:
 - espnet
 - audio
 - audio-to-audio
-language: en
 datasets:
-- universal_se
+- VCTK_DEMAND
+- DNS2020
+- WHAMR
+language: en
 license: cc-by-4.0
 ---
 
@@ -13,7 +15,7 @@ license: cc-by-4.0
 
 ### `wyz/vctk_dns2020_whamr_tfgridnet_tiny`
 
-This model was trained by wyz using universal_se recipe in [espnet](https://github.com/espnet/espnet/).
+This model was trained by wyz based on the universal_se_v1 recipe in [espnet](https://github.com/espnet/espnet/).
 
 ### Demo: How to use in ESPnet2
 
@@ -34,8 +36,8 @@ model = SeparateSpeech.from_pretrained(
 )
 # For loading a downloaded model
 # model = SeparateSpeech(
-#     train_config="exp_vctk_dns20_whamr/enh_tfgridnet_small_raw/config.yaml",
-#     model_file="exp_vctk_dns20_whamr/enh_tfgridnet_small_raw/xxxx.pth",
+#     train_config="exp_vctk_dns20_whamr/enh_tfgridnet_tiny_raw/config.yaml",
+#     model_file="exp_vctk_dns20_whamr/enh_tfgridnet_tiny_raw/xxxx.pth",
 #     normalize_output_wav=True,
 #     device="cuda",
 # )
@@ -67,17 +69,6 @@ enhanced = model(audio[None, :], fs=fs)[0]
 |reverb_et_simu_8ch_multich|2.00|92.14|10.87|10.87|0.00|-9.03|3.11|3.43|3.92|3.88|
 |whamr_tt_mix_single_reverb_max_16k|2.22|93.77|11.78|11.78|0.00|10.54|3.24|3.50|4.07|3.79|
 
-abc
-<!-- Generated by ./scripts/utils/show_enh_score.sh -->
-# RESULTS
-## Environments
-- date: `Mon Jan 29 22:17:30 EST 2024`
-- python version: `3.9.18 (main, Sep 11 2023, 13:41:44)  [GCC 11.2.0]`
-- espnet version: `espnet 202304`
-- pytorch version: `pytorch 1.13.1`
-- Git hash: ``
-  - Commit date: ``
-
 
 ## enhanced_test_48k
 

exp_vctk_dns20_whamr/enh_tfgridnet_small_raw/66epoch.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6012a39ab5999a1c948d8a1d132c1fac3de1c7f2dce32e09d71d3a902cc44d48
+size 6144607

exp_vctk_dns20_whamr/enh_tfgridnet_small_raw/config.yaml

@@ -1,9 +1,9 @@
-config: conf/tuning/tfgridnet_small.yaml
+config: conf/tfgridnet_tiny.yaml
 print_config: false
 log_level: INFO
 dry_run: false
 iterator_type: chunk
-output_dir: exp_vctk_dns20_whamr/enh_tfgridnet_small_raw
+output_dir: exp_vctk_dns20_whamr/enh_tfgridnet_tiny_raw
 ngpu: 1
 seed: 0
 num_workers: 4

exp_vctk_dns20_whamr/enh_tfgridnet_small_raw/latest.pth

@@ -0,0 +1 @@
+66epoch.pth

exp_vctk_dns20_whamr/enh_tfgridnet_small_raw/valid.loss.best.pth

@@ -0,0 +1 @@
+25epoch.pth

meta.yaml

@@ -1,8 +1,8 @@
 espnet: '202304'
 files:
-  model_file: exp_vctk_dns20_whamr/enh_tfgridnet_small_raw/25epoch.pth
+  model_file: exp_vctk_dns20_whamr/enh_tfgridnet_tiny_raw/25epoch.pth
 python: "3.8.16 (default, Mar  2 2023, 03:21:46) \n[GCC 11.2.0]"
 timestamp: 1723014155.859109
 torch: 2.0.1+cu118
 yaml_files:
-  train_config: exp_vctk_dns20_whamr/enh_tfgridnet_small_raw/config.yaml
+  train_config: exp_vctk_dns20_whamr/enh_tfgridnet_tiny_raw/config.yaml