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

config

This commit is contained in:
shahules786 2022-11-02 10:47:03 +05:30
parent 4badf64540
commit a3e488f101
5 changed files with 6 additions and 6 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 : WaveUnet - model : Demucs
- dataset : Vctk - dataset : DNS-2020
- optimizer : Adam - optimizer : Adam
- hyperparameters : default - hyperparameters : default
- trainer : default - trainer : default

2
enhancer/cli/train_config/dataset/DNS-2020.yaml
View File

@ -1,5 +1,5 @@
_target_: enhancer.data.dataset.EnhancerDataset _target_: enhancer.data.dataset.EnhancerDataset
root_dir : /scratch/c.sistc3/MS-SNSD/DNS30 root_dir : /scratch/c.sistc3/MS-SNSD/DNS15
name : dns-2020 name : dns-2020
duration : 5.0 duration : 5.0
stride : 1 stride : 1

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

@ -3,6 +3,6 @@ metric : [stoi,pesq,si-sdr]
lr : 0.003 lr : 0.003
ReduceLr_patience : 10 ReduceLr_patience : 10
Early_stop : False Early_stop : False
ReduceLr_factor : 0.1 ReduceLr_factor : 0.5
min_lr : 0.0000001 min_lr : 0.0000001
EarlyStopping_patience : 10 EarlyStopping_patience : 10

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

@ -25,7 +25,7 @@ limit_val_batches: 1.0
log_every_n_steps: 50 log_every_n_steps: 50
max_epochs: 200 max_epochs: 200
max_steps: -1 max_steps: -1
max_time: null max_time: 00:47:00:00
min_epochs: 1 min_epochs: 1
min_steps: null min_steps: null
move_metrics_to_cpu: False move_metrics_to_cpu: False

2
enhancer/data/dataset.py
View File

@ -133,7 +133,7 @@ class TaskDataset(pl.LightningDataModule):
speaker_index = rng.choice(possible_indices) speaker_index = rng.choice(possible_indices)
possible_indices.remove(speaker_index) possible_indices.remove(speaker_index)
speaker_name = all_speakers[speaker_index] speaker_name = all_speakers[speaker_index]
print(f"Selected f{speaker_name} for valid") print(f"Selected {speaker_name} for valid")
file_indices = [ file_indices = [
i i
for i, file in enumerate(data) for i, file in enumerate(data)