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

use waveunet

This commit is contained in:
shahules786 2022-09-29 16:11:45 +05:30
parent a0f70010f2
commit c1b67c1e3a
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cli/train_config/config.yaml
View File

@ -1,5 +1,5 @@
defaults: defaults:
- model : Demucs - model : WaveUnet
- dataset : Vctk - dataset : Vctk
- optimizer : Adam - optimizer : Adam
- hyperparameters : default - hyperparameters : default

2
enhancer/models/waveunet.py
View File

@ -70,6 +70,8 @@ class WaveUnet(Model):
loss: Union[str, List] = "mse", loss: Union[str, List] = "mse",
metric:Union[str,List] = "mse" metric:Union[str,List] = "mse"
): ):
duration = dataset.duration if isinstance(dataset,EnhancerDataset) else None
sampling_rate = sampling_rate if dataset is None else dataset.sampling_rate
super().__init__(num_channels=num_channels, super().__init__(num_channels=num_channels,
sampling_rate=sampling_rate,lr=lr, sampling_rate=sampling_rate,lr=lr,
dataset=dataset,duration=duration,loss=loss, metric=metric dataset=dataset,duration=duration,loss=loss, metric=metric