VCTK + DEMUCS

enhancer/cli/train_config/dataset/Vctk.yaml

@@ -4,8 +4,8 @@ root_dir : /scratch/c.sistc3/DS_10283_2791
 duration : 4.5
 stride : 0.5
 sampling_rate: 16000
-batch_size: 4
-valid_minutes : 1
+batch_size: 128
+valid_minutes : 15
 files:
   train_clean : clean_trainset_28spk_wav
   test_clean : clean_testset_wav

enhancer/cli/train_config/trainer/default.yaml

@@ -1,5 +1,5 @@
 _target_: pytorch_lightning.Trainer
-accelerator: auto
+accelerator: gpu
 accumulate_grad_batches: 1
 amp_backend: native
 auto_lr_find: True
@@ -9,7 +9,7 @@ benchmark: False
 check_val_every_n_epoch: 1
 detect_anomaly: False
 deterministic: False
-devices: 1
+devices: 2
 enable_checkpointing: True
 enable_model_summary: True
 enable_progress_bar: True
@@ -23,7 +23,7 @@ limit_test_batches: 1.0
 limit_train_batches: 1.0
 limit_val_batches: 1.0
 log_every_n_steps: 50
-max_epochs: 3
+max_epochs: 200
 max_steps: -1
 max_time: null
 min_epochs: 1
@@ -38,7 +38,7 @@ precision: 32
 profiler: null
 reload_dataloaders_every_n_epochs: 0
 replace_sampler_ddp: True
-strategy: null
+strategy: ddp
 sync_batchnorm: False
 tpu_cores: null
 track_grad_norm: -1