Waveunet w/o stride

enhancer/cli/train_config/config.yaml

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

enhancer/cli/train_config/dataset/Vctk.yaml

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

enhancer/cli/train_config/hyperparameters/default.yaml

@@ -1,6 +1,6 @@
-loss : mae
+loss : mse
 metric : [stoi,pesq,si-sdr]
-lr : 0.0003
+lr : 0.001
 ReduceLr_patience : 10
 ReduceLr_factor : 0.5
 min_lr : 0.000001

enhancer/cli/train_config/model/WaveUnet.yaml

@@ -1,5 +1,5 @@
 _target_: enhancer.models.waveunet.WaveUnet
 num_channels : 1
-depth : 12
+depth : 9
 initial_output_channels: 24
 sampling_rate : 16000

enhancer/cli/train_config/trainer/default.yaml

@@ -34,7 +34,7 @@ num_nodes: 1
 num_processes: 1
 num_sanity_val_steps: 2
 overfit_batches: 0.0
-precision: 16
+precision: 32
 profiler: null
 reload_dataloaders_every_n_epochs: 0
 replace_sampler_ddp: True