Demus + DNS

2022-10-22 12:07:45 +05:30 · 2022-10-22 12:07:45 +05:30 · 4cfe7a2463
parent f492e44e6b
commit 4cfe7a2463
3 changed files with 5 additions and 5 deletions
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@ -1,6 +1,6 @@
 defaults:
  - model : Demucs
-  - dataset : Vctk
+  - dataset : DNS-2020
  - optimizer : Adam
  - hyperparameters : default
  - trainer : default
--- a/enhancer/cli/train_config/dataset/DNS-2020.yaml
+++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml
@ -3,9 +3,9 @@ root_dir : /scratch/c.sistc3/MS-SNSD/DNS10
 name : dns-2020
 duration : 5.0
 sampling_rate: 16000
-batch_size: 64
-stride: 2
-valid_minutes: 30.0
+batch_size: 128
+stride: 1
+valid_minutes: 15.0
 files:
  train_clean : CleanSpeech_training
  test_clean : CleanSpeech_testing
--- a/enhancer/cli/train_config/trainer/default.yaml
+++ b/enhancer/cli/train_config/trainer/default.yaml
@ -23,7 +23,7 @@ limit_test_batches: 1.0
 limit_train_batches: 1.0
 limit_val_batches: 1.0
 log_every_n_steps: 100
-max_epochs: 400
+max_epochs: 200
 max_time: 00:47:00:00
 min_epochs: 1
 min_steps: null