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.

Sign up for GitHub

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
Open

Add implementations of USES2 speech enhancement models #5761

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
9699f00
Add implementations of USES2 speech enhancement models (Swin and Comp)
Emrys365 Apr 24, 2024
6f2d4df
Merge branch 'tse_recipe' of https://github.com/Emrys365/espnet into …
Emrys365 Apr 24, 2024
fd95a85
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 24, 2024
030a479
Merge branch 'master' of https://github.com/espnet/espnet into tse_re…
Emrys365 Jun 9, 2024
9d480c8
Add a recipe: egs2/urgent24/enh1
Emrys365 Jun 9, 2024
014a671
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 9, 2024
720e948
Revert
Emrys365 Jun 9, 2024
105045c
Revert
Emrys365 Jun 9, 2024
126a1b5
Add CI tests; Remove the dependency on timm
Emrys365 Jun 17, 2024
7d6abf1
Merge branch 'master' of https://github.com/espnet/espnet into tse_re…
Emrys365 Jun 17, 2024
bfd3000
Merge branch 'master' of https://github.com/espnet/espnet into tse_re…
Emrys365 Jul 14, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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
pre-commit-ci[bot] committed Apr 24, 2024
commit fd95a8558d3f845adf2b02caa2b564c1c3b6ac84
2 changes: 1 addition & 1 deletion espnet2/enh/layers/uses2_swin.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@
import torch
import torch.nn as nn

from espnet2.enh.layers.tcn import ChannelwiseLayerNorm
from espnet2.enh.layers.swin_transformer import BasicLayer
from espnet2.enh.layers.tcn import ChannelwiseLayerNorm
from espnet2.enh.layers.uses import ChannelAttention, ChannelTAC, LayerNormalization

EPS = torch.finfo(torch.float32).eps
Expand Down
2 changes: 1 addition & 1 deletion espnet2/tasks/enh.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -64,8 +64,8 @@
from espnet2.enh.separator.tfgridnet_separator import TFGridNet
from espnet2.enh.separator.tfgridnetv2_separator import TFGridNetV2
from espnet2.enh.separator.transformer_separator import TransformerSeparator
from espnet2.enh.separator.uses_separator import USESSeparator
from espnet2.enh.separator.uses2_separator import USES2Separator
from espnet2.enh.separator.uses_separator import USESSeparator
from espnet2.iterators.abs_iter_factory import AbsIterFactory
from espnet2.tasks.abs_task import AbsTask
from espnet2.torch_utils.initialize import initialize
Expand Down
Loading