Update model

README.md

@@ -0,0 +1,350 @@
+---
+tags:
+- espnet
+- audio
+- diarization
+language: noinfo
+datasets:
+- librimix
+license: cc-by-4.0
+---
+
+## ESPnet2 DIAR model 
+
+### `soumi-maiti/libri2mix_eend_ss`
+
+This model was trained by soumimaiti using librimix recipe in [espnet](https://github.com/espnet/espnet/).
+
+### Demo: How to use in ESPnet2
+
+Follow the [ESPnet installation instructions](https://espnet.github.io/espnet/installation.html)
+if you haven't done that already.
+
+```bash
+cd espnet
+git checkout d837c97c88f13ffe655a30bcff93d814f212b225
+pip install -e .
+cd egs2/librimix/enh_diar1
+./run.sh --skip_data_prep false --skip_train true --download_model soumi-maiti/libri2mix_eend_ss
+```
+
+
+
+## DIAR config
+
+<details><summary>expand</summary>
+
+```
+config: conf/tuning/train_diar_enh_convtasnet_concat_feats.yaml
+print_config: false
+log_level: INFO
+dry_run: false
+iterator_type: chunk
+output_dir: exp/diar_enh_train_diar_enh_convtasnet_concat_feats_raw
+ngpu: 1
+seed: 0
+num_workers: 4
+num_att_plot: 3
+dist_backend: nccl
+dist_init_method: env://
+dist_world_size: 2
+dist_rank: 0
+local_rank: 0
+dist_master_addr: localhost
+dist_master_port: 54493
+dist_launcher: null
+multiprocessing_distributed: true
+unused_parameters: false
+sharded_ddp: false
+cudnn_enabled: true
+cudnn_benchmark: false
+cudnn_deterministic: true
+collect_stats: false
+write_collected_feats: false
+max_epoch: 100
+patience: 50
+val_scheduler_criterion:
+- valid
+- loss
+early_stopping_criterion:
+- valid
+- loss
+- min
+best_model_criterion:
+-   - valid
+    - loss_enh
+    - min
+keep_nbest_models: 1
+nbest_averaging_interval: 0
+grad_clip: 5.0
+grad_clip_type: 2.0
+grad_noise: false
+accum_grad: 2
+no_forward_run: false
+resume: true
+train_dtype: float32
+use_amp: false
+log_interval: null
+use_matplotlib: true
+use_tensorboard: true
+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: 8
+valid_batch_size: null
+batch_bins: 1000000
+valid_batch_bins: null
+train_shape_file:
+- exp/diar_enh_stats_8k/train/speech_shape
+- exp/diar_enh_stats_8k/train/text_shape
+- exp/diar_enh_stats_8k/train/speech_ref1_shape
+- exp/diar_enh_stats_8k/train/speech_ref2_shape
+- exp/diar_enh_stats_8k/train/noise_ref1_shape
+valid_shape_file:
+- exp/diar_enh_stats_8k/valid/speech_shape
+- exp/diar_enh_stats_8k/valid/text_shape
+- exp/diar_enh_stats_8k/valid/speech_ref1_shape
+- exp/diar_enh_stats_8k/valid/speech_ref2_shape
+- exp/diar_enh_stats_8k/valid/noise_ref1_shape
+batch_type: folded
+valid_batch_type: null
+fold_length:
+- 800
+- 80000
+- 80000
+- 80000
+- 80000
+sort_in_batch: descending
+sort_batch: descending
+multiple_iterator: false
+chunk_length: 24000
+chunk_shift_ratio: 0.5
+num_cache_chunks: 1024
+train_data_path_and_name_and_type:
+-   - dump/raw/train/wav.scp
+    - speech
+    - sound
+-   - dump/raw/train/espnet_rttm
+    - text
+    - rttm
+-   - dump/raw/train/spk1.scp
+    - speech_ref1
+    - sound
+-   - dump/raw/train/spk2.scp
+    - speech_ref2
+    - sound
+-   - dump/raw/train/noise1.scp
+    - noise_ref1
+    - sound
+valid_data_path_and_name_and_type:
+-   - dump/raw/dev/wav.scp
+    - speech
+    - sound
+-   - dump/raw/dev/espnet_rttm
+    - text
+    - rttm
+-   - dump/raw/dev/spk1.scp
+    - speech_ref1
+    - sound
+-   - dump/raw/dev/spk2.scp
+    - speech_ref2
+    - sound
+-   - dump/raw/dev/noise1.scp
+    - noise_ref1
+    - 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-07
+    weight_decay: 0
+scheduler: reducelronplateau
+scheduler_conf:
+    mode: min
+    factor: 0.5
+    patience: 50
+token_list: null
+src_token_list: null
+init: xavier_uniform
+input_size: null
+ctc_conf:
+    dropout_rate: 0.0
+    ctc_type: builtin
+    reduce: true
+    ignore_nan_grad: null
+    zero_infinity: true
+enh_criterions:
+-   name: si_snr
+    conf:
+        eps: 1.0e-07
+    wrapper: pit
+    wrapper_conf:
+        weight: 1.0
+        independent_perm: true
+diar_num_spk: 2
+diar_input_size: 128
+enh_model_conf:
+    loss_type: si_snr
+asr_model_conf:
+    ctc_weight: 0.5
+    interctc_weight: 0.0
+    ignore_id: -1
+    lsm_weight: 0.0
+    length_normalized_loss: false
+    report_cer: true
+    report_wer: true
+    sym_space: <space>
+    sym_blank: <blank>
+    extract_feats_in_collect_stats: true
+st_model_conf:
+    stft_consistency: false
+    loss_type: mask_mse
+    mask_type: null
+diar_model_conf:
+    diar_weight: 0.2
+    attractor_weight: 0.2
+subtask_series:
+- enh
+- diar
+model_conf:
+    calc_enh_loss: true
+    bypass_enh_prob: 0
+use_preprocessor: true
+token_type: bpe
+bpemodel: null
+src_token_type: bpe
+src_bpemodel: null
+non_linguistic_symbols: null
+cleaner: null
+g2p: null
+enh_encoder: conv
+enh_encoder_conf:
+    channel: 512
+    kernel_size: 16
+    stride: 8
+enh_separator: tcn_nomask
+enh_separator_conf:
+    layer: 8
+    stack: 3
+    bottleneck_dim: 128
+    hidden_dim: 512
+    kernel: 3
+    causal: false
+    norm_type: gLN
+enh_decoder: conv
+enh_decoder_conf:
+    channel: 512
+    kernel_size: 16
+    stride: 8
+enh_mask_module: multi_mask
+enh_mask_module_conf:
+    max_num_spk: 3
+    mask_nonlinear: relu
+    bottleneck_dim: 128
+frontend: default
+frontend_conf: {}
+specaug: null
+specaug_conf: {}
+normalize: utterance_mvn
+normalize_conf: {}
+asr_preencoder: null
+asr_preencoder_conf: {}
+asr_encoder: rnn
+asr_encoder_conf: {}
+asr_postencoder: null
+asr_postencoder_conf: {}
+asr_decoder: rnn
+asr_decoder_conf: {}
+st_preencoder: null
+st_preencoder_conf: {}
+st_encoder: rnn
+st_encoder_conf: {}
+st_postencoder: null
+st_postencoder_conf: {}
+st_decoder: rnn
+st_decoder_conf: {}
+st_extra_asr_decoder: rnn
+st_extra_asr_decoder_conf: {}
+st_extra_mt_decoder: rnn
+st_extra_mt_decoder_conf: {}
+diar_frontend: default
+diar_frontend_conf:
+    hop_length: 64
+    fs: 8000
+diar_specaug: null
+diar_specaug_conf: {}
+diar_normalize: utterance_mvn
+diar_normalize_conf: {}
+diar_encoder: transformer
+diar_encoder_conf:
+    input_layer: conv2d8
+    num_blocks: 4
+    linear_units: 512
+    dropout_rate: 0.1
+    output_size: 256
+    attention_heads: 4
+    attention_dropout_rate: 0.1
+diar_decoder: linear
+diar_decoder_conf: {}
+label_aggregator: label_aggregator
+label_aggregator_conf:
+    win_length: 256
+    hop_length: 64
+diar_attractor: rnn
+diar_attractor_conf:
+    unit: 256
+    layer: 1
+    dropout: 0.0
+    attractor_grad: true
+required:
+- output_dir
+version: '202205'
+distributed: true
+```
+
+</details>
+
+
+
+### Citing ESPnet
+
+```BibTex
+@inproceedings{watanabe2018espnet,
+  author={Shinji Watanabe and Takaaki Hori and Shigeki Karita and Tomoki Hayashi and Jiro Nishitoba and Yuya Unno and Nelson Yalta and Jahn Heymann and Matthew Wiesner and Nanxin Chen and Adithya Renduchintala and Tsubasa Ochiai},
+  title={{ESPnet}: End-to-End Speech Processing Toolkit},
+  year={2018},
+  booktitle={Proceedings of Interspeech},
+  pages={2207--2211},
+  doi={10.21437/Interspeech.2018-1456},
+  url={http://dx.doi.org/10.21437/Interspeech.2018-1456}
+}
+
+
+
+
+```
+
+or arXiv:
+
+```bibtex
+@misc{watanabe2018espnet,
+  title={ESPnet: End-to-End Speech Processing Toolkit}, 
+  author={Shinji Watanabe and Takaaki Hori and Shigeki Karita and Tomoki Hayashi and Jiro Nishitoba and Yuya Unno and Nelson Yalta and Jahn Heymann and Matthew Wiesner and Nanxin Chen and Adithya Renduchintala and Tsubasa Ochiai},
+  year={2018},
+  eprint={1804.00015},
+  archivePrefix={arXiv},
+  primaryClass={cs.CL}
+}
+```

exp/diar_enh_stats_8k/train/feats_stats.npz

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

exp/diar_enh_train_diar_enh_convtasnet_concat_feats_raw/57epoch.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8adef049fd941d6deb198a9dbfe05b4f8dd702a3d3ff496ae197c5608fce80ea
+size 38983318

exp/diar_enh_train_diar_enh_convtasnet_concat_feats_raw/DIAR_RESULTS.md

@@ -0,0 +1,15 @@
+<!-- Generated by scripts/utils/show_diar_result.sh -->
+# RESULTS
+## Environments
+- date: `Thu Sep  1 06:42:28 EDT 2022`
+- python version: `3.9.7 (default, Sep 16 2021, 13:09:58)  [GCC 7.5.0]`
+- espnet version: `espnet 202205`
+- pytorch version: `pytorch 1.8.1+cu102`
+- Git hash: `d837c97c88f13ffe655a30bcff93d814f212b225`
+  - Commit date: `Wed Jun 29 12:04:57 2022 -0700`
+
+## diar_enh_train_diar_enh_convtasnet_concat_feats_raw
+### DER
+diarized_enhanced_test_lat_enh
+|threshold_median_collar|DER|
+|---|---|

exp/diar_enh_train_diar_enh_convtasnet_concat_feats_raw/ENH_RESULTS.md

@@ -0,0 +1,35 @@
+<!-- Generated by scripts/utils/show_enh_score.sh -->
+# RESULTS
+## Environments
+- date: `Thu Sep  1 06:50:40 EDT 2022`
+- python version: `3.9.7 (default, Sep 16 2021, 13:09:58)  [GCC 7.5.0]`
+- espnet version: `espnet 202205`
+- pytorch version: `pytorch 1.8.1+cu102`
+- Git hash: `d837c97c88f13ffe655a30bcff93d814f212b225`
+  - Commit date: `Wed Jun 29 12:04:57 2022 -0700`
+
+
+## diar_enh_train_diar_enh_convtasnet_concat_feats_raw
+
+config: conf/tuning/train_diar_enh_convtasnet_concat_feats.yaml
+
+|dataset|STOI|SAR|SDR|SIR|SI_SNR|
+|---|---|---|---|---|---|
+|diarized_enhanced_dev|83.6928|10.6693|10.1626|21.8855|9.4331|
+|diarized_enhanced_dev_wo_diar|83.7127|10.6663|10.1550|21.8520|9.4306|
+|diarized_enhanced_test|83.7409|10.5066|9.9290|21.3136|9.1342|
+|diarized_enhanced_test_ave_enh|83.7409|10.5066|9.9290|21.3136|9.1342|
+|diarized_enhanced_test_ave_enh2|83.7560|10.5096|9.9249|21.2740|9.1355|
+|diarized_enhanced_test_decode_diar|83.7560|10.5096|9.9249|21.2740|9.1355|
+|diarized_enhanced_test_decode_diar_enh|83.7409|10.5066|9.9290|21.3136|9.1342|
+|diarized_enhanced_test_decode_diar_enh2|83.7560|10.5096|9.9249|21.2740|9.1355|
+|diarized_enhanced_test_lat_enh|83.7105|10.5050|9.9230|21.3135|9.1285|
+|diarized_enhanced_test_lat_enh2|83.7299|10.5073|9.9189|21.2735|9.1314|
+|diarized_enhanced_test_sparse_2_0.2_decode_diar_enh|86.6907|11.2629|10.9169|28.3233|10.0191|
+|diarized_enhanced_test_sparse_2_0.4_decode_diar_enh|85.3315|10.7804|10.3049|24.7652|9.4069|
+|diarized_enhanced_test_sparse_2_0.6_decode_diar_enh|84.6040|10.5808|10.0880|23.3730|9.2251|
+|diarized_enhanced_test_sparse_2_0.8_decode_diar_enh|84.1088|10.3668|9.8149|22.3321|8.9487|
+|diarized_enhanced_test_sparse_2_0_decode_diar_enh|87.6317|11.7109|11.3008|54.0680|10.3340|
+|diarized_enhanced_test_sparse_2_1_decode_diar_enh|84.2412|10.2397|9.6977|21.8898|8.8254|
+|diarized_enhanced_test_wo_diar|83.7560|10.5096|9.9249|21.2740|9.1355|
+

exp/diar_enh_train_diar_enh_convtasnet_concat_feats_raw/config.yaml

@@ -0,0 +1,277 @@
+config: conf/tuning/train_diar_enh_convtasnet_concat_feats.yaml
+print_config: false
+log_level: INFO
+dry_run: false
+iterator_type: chunk
+output_dir: exp/diar_enh_train_diar_enh_convtasnet_concat_feats_raw
+ngpu: 1
+seed: 0
+num_workers: 4
+num_att_plot: 3
+dist_backend: nccl
+dist_init_method: env://
+dist_world_size: 2
+dist_rank: 0
+local_rank: 0
+dist_master_addr: localhost
+dist_master_port: 54493
+dist_launcher: null
+multiprocessing_distributed: true
+unused_parameters: false
+sharded_ddp: false
+cudnn_enabled: true
+cudnn_benchmark: false
+cudnn_deterministic: true
+collect_stats: false
+write_collected_feats: false
+max_epoch: 100
+patience: 50
+val_scheduler_criterion:
+- valid
+- loss
+early_stopping_criterion:
+- valid
+- loss
+- min
+best_model_criterion:
+-   - valid
+    - loss_enh
+    - min
+keep_nbest_models: 1
+nbest_averaging_interval: 0
+grad_clip: 5.0
+grad_clip_type: 2.0
+grad_noise: false
+accum_grad: 2
+no_forward_run: false
+resume: true
+train_dtype: float32
+use_amp: false
+log_interval: null
+use_matplotlib: true
+use_tensorboard: true
+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: 8
+valid_batch_size: null
+batch_bins: 1000000
+valid_batch_bins: null
+train_shape_file:
+- exp/diar_enh_stats_8k/train/speech_shape
+- exp/diar_enh_stats_8k/train/text_shape
+- exp/diar_enh_stats_8k/train/speech_ref1_shape
+- exp/diar_enh_stats_8k/train/speech_ref2_shape
+- exp/diar_enh_stats_8k/train/noise_ref1_shape
+valid_shape_file:
+- exp/diar_enh_stats_8k/valid/speech_shape
+- exp/diar_enh_stats_8k/valid/text_shape
+- exp/diar_enh_stats_8k/valid/speech_ref1_shape
+- exp/diar_enh_stats_8k/valid/speech_ref2_shape
+- exp/diar_enh_stats_8k/valid/noise_ref1_shape
+batch_type: folded
+valid_batch_type: null
+fold_length:
+- 800
+- 80000
+- 80000
+- 80000
+- 80000
+sort_in_batch: descending
+sort_batch: descending
+multiple_iterator: false
+chunk_length: 24000
+chunk_shift_ratio: 0.5
+num_cache_chunks: 1024
+train_data_path_and_name_and_type:
+-   - dump/raw/train/wav.scp
+    - speech
+    - sound
+-   - dump/raw/train/espnet_rttm
+    - text
+    - rttm
+-   - dump/raw/train/spk1.scp
+    - speech_ref1
+    - sound
+-   - dump/raw/train/spk2.scp
+    - speech_ref2
+    - sound
+-   - dump/raw/train/noise1.scp
+    - noise_ref1
+    - sound
+valid_data_path_and_name_and_type:
+-   - dump/raw/dev/wav.scp
+    - speech
+    - sound
+-   - dump/raw/dev/espnet_rttm
+    - text
+    - rttm
+-   - dump/raw/dev/spk1.scp
+    - speech_ref1
+    - sound
+-   - dump/raw/dev/spk2.scp
+    - speech_ref2
+    - sound
+-   - dump/raw/dev/noise1.scp
+    - noise_ref1
+    - 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-07
+    weight_decay: 0
+scheduler: reducelronplateau
+scheduler_conf:
+    mode: min
+    factor: 0.5
+    patience: 50
+token_list: null
+src_token_list: null
+init: xavier_uniform
+input_size: null
+ctc_conf:
+    dropout_rate: 0.0
+    ctc_type: builtin
+    reduce: true
+    ignore_nan_grad: null
+    zero_infinity: true
+enh_criterions:
+-   name: si_snr
+    conf:
+        eps: 1.0e-07
+    wrapper: pit
+    wrapper_conf:
+        weight: 1.0
+        independent_perm: true
+diar_num_spk: 2
+diar_input_size: 128
+enh_model_conf:
+    loss_type: si_snr
+asr_model_conf:
+    ctc_weight: 0.5
+    interctc_weight: 0.0
+    ignore_id: -1
+    lsm_weight: 0.0
+    length_normalized_loss: false
+    report_cer: true
+    report_wer: true
+    sym_space: <space>
+    sym_blank: <blank>
+    extract_feats_in_collect_stats: true
+st_model_conf:
+    stft_consistency: false
+    loss_type: mask_mse
+    mask_type: null
+diar_model_conf:
+    diar_weight: 0.2
+    attractor_weight: 0.2
+subtask_series:
+- enh
+- diar
+model_conf:
+    calc_enh_loss: true
+    bypass_enh_prob: 0
+use_preprocessor: true
+token_type: bpe
+bpemodel: null
+src_token_type: bpe
+src_bpemodel: null
+non_linguistic_symbols: null
+cleaner: null
+g2p: null
+enh_encoder: conv
+enh_encoder_conf:
+    channel: 512
+    kernel_size: 16
+    stride: 8
+enh_separator: tcn_nomask
+enh_separator_conf:
+    layer: 8
+    stack: 3
+    bottleneck_dim: 128
+    hidden_dim: 512
+    kernel: 3
+    causal: false
+    norm_type: gLN
+enh_decoder: conv
+enh_decoder_conf:
+    channel: 512
+    kernel_size: 16
+    stride: 8
+enh_mask_module: multi_mask
+enh_mask_module_conf:
+    max_num_spk: 3
+    mask_nonlinear: relu
+    bottleneck_dim: 128
+frontend: default
+frontend_conf: {}
+specaug: null
+specaug_conf: {}
+normalize: utterance_mvn
+normalize_conf: {}
+asr_preencoder: null
+asr_preencoder_conf: {}
+asr_encoder: rnn
+asr_encoder_conf: {}
+asr_postencoder: null
+asr_postencoder_conf: {}
+asr_decoder: rnn
+asr_decoder_conf: {}
+st_preencoder: null
+st_preencoder_conf: {}
+st_encoder: rnn
+st_encoder_conf: {}
+st_postencoder: null
+st_postencoder_conf: {}
+st_decoder: rnn
+st_decoder_conf: {}
+st_extra_asr_decoder: rnn
+st_extra_asr_decoder_conf: {}
+st_extra_mt_decoder: rnn
+st_extra_mt_decoder_conf: {}
+diar_frontend: default
+diar_frontend_conf:
+    hop_length: 64
+    fs: 8000
+diar_specaug: null
+diar_specaug_conf: {}
+diar_normalize: utterance_mvn
+diar_normalize_conf: {}
+diar_encoder: transformer
+diar_encoder_conf:
+    input_layer: conv2d8
+    num_blocks: 4
+    linear_units: 512
+    dropout_rate: 0.1
+    output_size: 256
+    attention_heads: 4
+    attention_dropout_rate: 0.1
+diar_decoder: linear
+diar_decoder_conf: {}
+label_aggregator: label_aggregator
+label_aggregator_conf:
+    win_length: 256
+    hop_length: 64
+diar_attractor: rnn
+diar_attractor_conf:
+    unit: 256
+    layer: 1
+    dropout: 0.0
+    attractor_grad: true
+required:
+- output_dir
+version: '202205'
+distributed: true

meta.yaml

@@ -0,0 +1,8 @@
+espnet: '202205'
+files:
+  model_file: exp/diar_enh_train_diar_enh_convtasnet_concat_feats_raw/57epoch.pth
+python: "3.9.7 (default, Sep 16 2021, 13:09:58) \n[GCC 7.5.0]"
+timestamp: 1676313507.836365
+torch: 1.8.1+cu102
+yaml_files:
+  train_config: exp/diar_enh_train_diar_enh_convtasnet_concat_feats_raw/config.yaml