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

config

This commit is contained in:
shahules786 2022-10-20 09:53:27 +05:30
parent c5824cb34a
commit f2561d7cf7
3 changed files with 5 additions and 4 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

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

@ -1,9 +1,10 @@
_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 : 1.0 duration : 4.5
stride : 0.5
sampling_rate: 16000 sampling_rate: 16000
batch_size: 128 batch_size: 16
files: files:
train_clean : clean_trainset_28spk_wav train_clean : clean_trainset_28spk_wav

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

@ -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