demucs 250

2022-10-14 12:45:34 +05:30 · 2022-10-14 12:45:34 +05:30 · 0e58691a2c
parent 807f4b93ea
commit 0e58691a2c
3 changed files with 3 additions and 3 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,7 +1,7 @@
 _target_: enhancer.data.dataset.EnhancerDataset
 name : vctk
 root_dir : /scratch/c.sistc3/DS_10283_2791
-duration : 1.0
+duration : 1.5
 sampling_rate: 16000
 batch_size: 256
 valid_size : 0.05
--- 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: 500
+max_epochs: 250
 max_time: null
 min_epochs: 1
 min_steps: null