Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dccrn

This commit is contained in:
shahules786 2022-11-08 17:18:45 +05:30
parent c4e392aff5
commit cad0bbedc8
4 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
enhancer/cli/train_config/config.yaml
View File

@ -1,6 +1,6 @@
defaults: defaults:
- model : Demucs - model : DCCRN
- dataset : DNS-2020 - dataset : Vctk
- optimizer : Adam - optimizer : Adam
- hyperparameters : default - hyperparameters : default
- trainer : default - trainer : default

2
enhancer/cli/train_config/hyperparameters/default.yaml
View File

@ -1,6 +1,6 @@
loss : si-snr loss : si-snr
metric : [stoi,pesq] metric : [stoi,pesq]
lr : 0.0003 lr : 0.001
ReduceLr_patience : 10 ReduceLr_patience : 10
Early_stop : False Early_stop : False
ReduceLr_factor : 0.2 ReduceLr_factor : 0.2

2
enhancer/cli/train_config/mlflow/experiment.yaml
View File

@ -1,2 +1,2 @@
experiment_name : shahules/enhancer experiment_name : shahules/enhancer
run_name : Demucs-DNS30 run_name : DCCRN-Vctk28

1
enhancer/utils/transforms.py
View File

@ -86,6 +86,7 @@ class ConviSTFT(ConvFFT):
out = F.conv_transpose1d(input, self.weight, stride=self.hop_size) out = F.conv_transpose1d(input, self.weight, stride=self.hop_size)
coeff = self.window.unsqueeze(1).repeat(1, 1, input.size(-1)) ** 2 coeff = self.window.unsqueeze(1).repeat(1, 1, input.size(-1)) ** 2
coeff.to(input.device) coeff.to(input.device)
self.enframe.to(input.device)
coeff = F.conv_transpose1d(coeff, self.enframe, stride=self.hop_size) coeff = F.conv_transpose1d(coeff, self.enframe, stride=self.hop_size)
out = out / (coeff + 1e-8) out = out / (coeff + 1e-8)
pad = self.window_len - self.hop_size pad = self.window_len - self.hop_size