DNS-2020 + VTCK

enhancer/cli/train_config/config.yaml

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

enhancer/cli/train_config/dataset/Vctk.yaml

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

enhancer/cli/train_config/trainer/default.yaml

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