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

config

This commit is contained in:
shahules786 2022-10-27 15:30:06 +05:30
parent dbfa580618
commit 77f3658e5c
5 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,10 +2,10 @@ _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 : 4.5 duration : 4.5
stride : 2 stride : 0.5
sampling_rate: 16000 sampling_rate: 16000
batch_size: 32 batch_size: 32
valid_minutes : 15 valid_minutes : 25
files: files:
train_clean : clean_trainset_28spk_wav train_clean : clean_trainset_28spk_wav
test_clean : clean_testset_wav test_clean : clean_testset_wav

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

@ -1,7 +1,8 @@
loss : mae loss : mae
metric : [stoi,pesq,si-sdr] metric : [stoi,pesq,si-sdr]
lr : 0.0003 lr : 0.0003
ReduceLr_patience : 5 ReduceLr_patience : 10
ReduceLr_factor : 0.2 Early_stop : True
ReduceLr_factor : 0.5
min_lr : 0.000001 min_lr : 0.000001
EarlyStopping_factor : 10 EarlyStopping_patience : 10

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 + Vtck with stride + augmentations run_name : Demucs(ablation_study)

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

@ -23,7 +23,7 @@ limit_test_batches: 1.0
limit_train_batches: 1.0 limit_train_batches: 1.0
limit_val_batches: 1.0 limit_val_batches: 1.0
log_every_n_steps: 50 log_every_n_steps: 50
max_epochs: 200 max_epochs: 250
max_steps: -1 max_steps: -1
max_time: null max_time: null
min_epochs: 1 min_epochs: 1

1
enhancer/data/dataset.py
View File

@ -136,6 +136,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")
file_indices = [ file_indices = [
i i
for i, file in enumerate(data) for i, file in enumerate(data)