Siddhant's picture
Add audio-to-audio tag in audio-source-separation models
771d692
metadata
tags:
  - espnet
  - audio
  - audio-source-separation
  - audio-to-audio
language: en
datasets:
  - wsj0_2mix
license: cc-by-4.0
inference: false

ESPnet2 ENH pretrained model

Chenda Li/wsj0_2mix_enh_train_enh_rnn_tf_raw_valid.si_snr.ave, fs=8k, lang=en

♻️ Imported from https://zenodo.org/record/4498554#.YOAOEpozZH4.

This model was trained by Chenda Li using wsj0_2mix recipe in espnet.

Python API

See https://github.com/espnet/espnet_model_zoo

Evaluate in the recipe

# coming soon

Results

# RESULTS
## Environments
- date: `Thu Feb  4 01:08:19 CST 2021`
- python version: `3.7.6 (default, Jan  8 2020, 19:59:22)  [GCC 7.3.0]`
- espnet version: `espnet 0.9.7`
- pytorch version: `pytorch 1.5.0`
- Git hash: `a3334220b0352931677946d178fade3313cf82bb`
  - Commit date: `Fri Jan 29 23:35:47 2021 +0800`


## enh_train_enh_rnn_tf_raw

config: conf/tuning/train_enh_rnn_tf.yaml

|dataset|STOI|SAR|SDR|SIR|
|---|---|---|---|---|
|enhanced_cv_min_8k|0.891065|11.556|10.3982|18.0655|
|enhanced_tt_min_8k|0.896373|11.4086|10.2433|18.0496|

Training config

See full config in config.yaml

config: conf/tuning/train_enh_rnn_tf.yaml
print_config: false
log_level: INFO
dry_run: false
iterator_type: sequence
output_dir: exp/enh_train_enh_rnn_tf_raw
ngpu: 1
seed: 0
num_workers: 4
num_att_plot: 3
dist_backend: nccl
dist_init_method: env://
dist_world_size: null
dist_rank: null
local_rank: 0
dist_master_addr: null
dist_master_port: null
dist_launcher: null
multiprocessing_distributed: false
cudnn_enabled: true
cudnn_benchmark: false
cudnn_deterministic: true