ESPnet2 pretrained model, Zhong-Qiu Wang/wsj0_2mix_enh_train_enh_tfgridnet_tf_lr-patience3_patience5_raw_valid.loss.ave, fs=8k, lang=en
Creators: name: Wang, Zhong-Qiu
Publisher: name: Zenodo
Year: 2023
Subjects: ESPnet
deep-learning
python
pytorch
speech-recognition
speech-synthesis
speech-translation
machine-translation

description: This model was trained by Zhong-Qiu Wang using wsj0_2mix recipe in espnet. <strong>Python API</strong><pre><code class="language-python">See https://github.com/espnet/espnet_model_zoo</code></pre> <strong>Evaluate in the recipe</strong><pre><code class="language-bash">git clone https://github.com/espnet/espnet cd espnet git checkout 8d154cdf41edff404c5eca662f485868cd0c4f36 pip install -e . cd egs2/wsj0_2mix/enh1 ./run.sh --skip_data_prep false --skip_train true --download_model Zhong-Qiu Wang/wsj0_2mix_enh_train_enh_tfgridnet_tf_lr-patience3_patience5_raw_valid.loss.ave</code> </pre> <strong>Results</strong><pre><code> # RESULTS ## Environments - date: `Sun Jan 8 15:55:53 EST 2023` - python version: `3.8.15 (default, Nov 24 2022, 15:19:38) [GCC 11.2.0]` - espnet version: `espnet 202207` - pytorch version: `pytorch 1.9.0` - Git hash: `83319022d773c4e6dc075be2b863103d2c48fe97` - Commit date: `Thu Aug 18 17:02:21 2022 +0900` ## enh_train_enh_tfgridnet_tf_lr-patience3_patience5_raw config: conf/tuning/train_enh_tfgridnet_tf_lr-patience3_patience5.yaml |dataset|STOI|SAR|SDR|SIR|SI_SNR| |---|---|---|---|---|---| |enhanced_cv_min_8k|97.98|22.62|22.29|34.54|22.24| |enhanced_tt_min_8k|98.68|23.23|22.90|35.30|22.81|</code></pre> <strong>ASR config</strong><pre><code>config: conf/tuning/train_enh_tfgridnet_tf_lr-patience3_patience5.yaml print_config: false log_level: INFO dry_run: false iterator_type: chunk output_dir: exp/enh_train_enh_tfgridnet_tf_lr-patience3_patience5_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 unused_parameters: false sharded_ddp: false cudnn_enabled: true cudnn_benchmark: false cudnn_deterministic: true collect_stats: false write_collected_feats: false max_epoch: 150 patience: 5 val_scheduler_criterion: - valid - loss early_stopping_criterion: - valid - loss - min best_model_criterion: - - valid - si_snr - max - - valid - loss - min keep_nbest_models: 1 nbest_averaging_interval: 0 grad_clip: 5.0 grad_clip_type: 2.0 grad_noise: false accum_grad: 1 no_forward_run: false resume: true train_dtype: float32 use_amp: false log_interval: null use_matplotlib: true use_tensorboard: true create_graph_in_tensorboard: false use_wandb: false wandb_project: null wandb_id: null wandb_entity: null wandb_name: null wandb_model_log_interval: -1 detect_anomaly: false pretrain_path: null init_param: [] ignore_init_mismatch: false freeze_param: [] num_iters_per_epoch: null batch_size: 3 valid_batch_size: null batch_bins: 1000000 valid_batch_bins: null train_shape_file: - exp/enh_stats_8k/train/speech_mix_shape - exp/enh_stats_8k/train/speech_ref1_shape - exp/enh_stats_8k/train/speech_ref2_shape valid_shape_file: - exp/enh_stats_8k/valid/speech_mix_shape - exp/enh_stats_8k/valid/speech_ref1_shape - exp/enh_stats_8k/valid/speech_ref2_shape batch_type: folded valid_batch_type: null fold_length: - 80000 - 80000 - 80000 sort_in_batch: descending sort_batch: descending multiple_iterator: false chunk_length: 32000 chunk_shift_ratio: 0.5 num_cache_chunks: 1024 train_data_path_and_name_and_type: - - dump/raw/tr_min_8k/wav.scp - speech_mix - sound - - dump/raw/tr_min_8k/spk1.scp - speech_ref1 - sound - - dump/raw/tr_min_8k/spk2.scp - speech_ref2 - sound valid_data_path_and_name_and_type: - - dump/raw/cv_min_8k/wav.scp - speech_mix - sound - - dump/raw/cv_min_8k/spk1.scp - speech_ref1 - sound - - dump/raw/cv_min_8k/spk2.scp - speech_ref2 - sound allow_variable_data_keys: false max_cache_size: 0.0 max_cache_fd: 32 valid_max_cache_size: null optim: adam optim_conf: lr: 0.001 eps: 1.0e-08 weight_decay: 0 scheduler: reducelronplateau scheduler_conf: mode: min factor: 0.5 patience: 3 init: xavier_uniform model_conf: stft_consistency: false loss_type: mask_mse mask_type: null criterions: - name: si_snr conf: eps: 1.0e-07 wrapper: pit wrapper_conf: weight: 1.0 independent_perm: true use_preprocessor: false speech_volume_normalize: null rir_scp: null rir_apply_prob: 1.0 noise_scp: null noise_apply_prob: 1.0 noise_db_range: '13_15' short_noise_thres: 0.5 use_reverberant_ref: false num_spk: 1 num_noise_type: 1 sample_rate: 8000 force_single_channel: false dynamic_mixing: false utt2spk: null dynamic_mixing_gain_db: 0.0 encoder: same encoder_conf: {} separator: tfgridnet separator_conf: n_srcs: 2 n_fft: 128 stride: 64 window: hann n_imics: 1 n_layers: 6 lstm_hidden_units: 192 attn_n_head: 4 attn_approx_qk_dim: 512 emb_dim: 48 emb_ks: 4 emb_hs: 1 activation: prelu eps: 1.0e-05 decoder: same decoder_conf: {} mask_module: multi_mask mask_module_conf: {} required: - output_dir version: '202207' distributed: false</code></pre>