waveunet

2022-10-31 10:11:22 +05:30 · 2022-10-31 10:11:22 +05:30 · 9e315ca6c4
parent 00bb38c95b
commit 9e315ca6c4
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 : Demucs
+  - model : WaveUnet
  - dataset : Vctk
  - optimizer : Adam
  - hyperparameters : default
--- a/enhancer/cli/train_config/dataset/Vctk.yaml
+++ b/enhancer/cli/train_config/dataset/Vctk.yaml
@ -4,7 +4,7 @@ root_dir : /scratch/c.sistc3/DS_10283_2791
 duration : 4.5
 stride : 0.5
 sampling_rate: 16000
-batch_size: 32
+batch_size: 128
 valid_minutes : 25
 files:
  train_clean : clean_trainset_28spk_wav
--- a/enhancer/cli/train_config/mlflow/experiment.yaml
+++ b/enhancer/cli/train_config/mlflow/experiment.yaml
@ -1,2 +1,2 @@
 experiment_name : shahules/enhancer
-run_name : Demucs-DNS(ablation_study-noaug)
+run_name : WaveUnet-vctk