Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add implementations of USES2 speech enhancement models #5761

Open
wants to merge 11 commits into
base: master
Choose a base branch
from
Prev Previous commit
Next Next commit
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jun 9, 2024
commit 014a671d8e75b745b47311a202af483649f2046f
Original file line number Diff line number Diff line change
Expand Up @@ -82,9 +82,9 @@ separator_conf:
# The overlall loss in the multi-task learning will be:
# loss = weight_1 * loss_1 + ... + weight_N * loss_N
# The default `weight` for each sub-loss is 1.0
criterions:
criterions:
# The first criterion
- name: mr_l1_tfd
- name: mr_l1_tfd
conf:
window_sz: [256, 512, 768, 1024]
hop_sz: null
Expand All @@ -95,7 +95,7 @@ criterions:
wrapper_conf:
weight: 1.0
# The second criterion
- name: si_snr
- name: si_snr
conf:
eps: 1.0e-7
wrapper: fixed_order
Expand Down
6 changes: 3 additions & 3 deletions egs2/urgent24/enh1/conf/tuning/train_enh_conv_tasnet.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -83,9 +83,9 @@ separator_conf:
# The overlall loss in the multi-task learning will be:
# loss = weight_1 * loss_1 + ... + weight_N * loss_N
# The default `weight` for each sub-loss is 1.0
criterions:
criterions:
# The first criterion
- name: mr_l1_tfd
- name: mr_l1_tfd
conf:
window_sz: [256, 512, 768, 1024]
hop_sz: null
Expand All @@ -96,7 +96,7 @@ criterions:
wrapper_conf:
weight: 1.0
# The second criterion
- name: si_snr
- name: si_snr
conf:
eps: 1.0e-7
wrapper: fixed_order
Expand Down
6 changes: 3 additions & 3 deletions egs2/urgent24/enh1/conf/tuning/train_enh_tfgridnet.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -87,9 +87,9 @@ separator_conf:
# The overlall loss in the multi-task learning will be:
# loss = weight_1 * loss_1 + ... + weight_N * loss_N
# The default `weight` for each sub-loss is 1.0
criterions:
criterions:
# The first criterion
- name: mr_l1_tfd
- name: mr_l1_tfd
conf:
window_sz: [256, 512, 768, 1024]
hop_sz: null
Expand All @@ -100,7 +100,7 @@ criterions:
wrapper_conf:
weight: 1.0
# The second criterion
- name: si_snr
- name: si_snr
conf:
eps: 1.0e-7
wrapper: fixed_order
Expand Down
Loading