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

DNS-2020 + VTCK

This commit is contained in:
shahules786 2022-10-22 11:20:25 +05:30
parent ed30840f4e
commit 4b0e8a5ef1
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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

6
enhancer/cli/train_config/dataset/Vctk.yaml
View File

@ -1,11 +1,11 @@
_target_: enhancer.data.dataset.EnhancerDataset _target_: enhancer.data.dataset.EnhancerDataset
name : vctk name : vctk
root_dir : /scratch/c.sistc3/DS_10283_2791 root_dir : /scratch/c.sistc3/DS_10283_2791
duration : 2 duration : 4.5
stride : 1 stride : 0.5
sampling_rate: 16000 sampling_rate: 16000
batch_size: 128 batch_size: 128
valid_minutes : 20 valid_minutes : 10
files: files:
train_clean : clean_trainset_28spk_wav train_clean : clean_trainset_28spk_wav

4
enhancer/cli/train_config/trainer/default.yaml
View File

@ -1,5 +1,5 @@
_target_: pytorch_lightning.Trainer _target_: pytorch_lightning.Trainer
accelerator: cpu accelerator: gpu
accumulate_grad_batches: 1 accumulate_grad_batches: 1
amp_backend: native amp_backend: native
auto_lr_find: False auto_lr_find: False
@ -9,7 +9,7 @@ benchmark: False
check_val_every_n_epoch: 1 check_val_every_n_epoch: 1
detect_anomaly: False detect_anomaly: False
deterministic: False deterministic: False
devices: 1 devices: -1
enable_checkpointing: True enable_checkpointing: True
enable_model_summary: True enable_model_summary: True
enable_progress_bar: True enable_progress_bar: True