config

2022-10-20 09:53:27 +05:30 · 2022-10-20 09:53:27 +05:30 · f2561d7cf7
parent c5824cb34a
commit f2561d7cf7
3 changed files with 5 additions and 4 deletions
--- a/enhancer/cli/train_config/config.yaml
+++ b/enhancer/cli/train_config/config.yaml
@ -1,5 +1,5 @@
 defaults:
-  - model : WaveUnet
+  - model : Demucs
  - dataset : Vctk
  - optimizer : Adam
  - hyperparameters : default
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/cli/train_config/dataset/Vctk.yaml
@ -1,9 +1,10 @@
 _target_: enhancer.data.dataset.EnhancerDataset
 name : vctk
 root_dir : /scratch/c.sistc3/DS_10283_2791
-duration : 1.0
+duration : 4.5
+stride : 0.5
 sampling_rate: 16000
-batch_size: 128
+batch_size: 16

 files:
  train_clean : clean_trainset_28spk_wav
--- a/enhancer/cli/train_config/trainer/default.yaml
+++ b/enhancer/cli/train_config/trainer/default.yaml
@ -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