reduce params

2022-09-30 10:55:39 +05:30 · 2022-09-30 10:55:39 +05:30 · 9170666592
parent ba271f8a2a
commit 9170666592
2 changed files with 3 additions and 3 deletions
--- a/cli/train_config/config.yaml
+++ b/cli/train_config/config.yaml
@ -1,5 +1,5 @@
 defaults:
-  - model : WaveUnet
+  - model : Demucs
  - dataset : Vctk
  - optimizer : Adam
  - hyperparameters : default
--- a/cli/train_config/model/Demucs.yaml
+++ b/cli/train_config/model/Demucs.yaml
@ -1,11 +1,11 @@
 _target_: enhancer.models.demucs.Demucs
 num_channels: 1
-resample: 4
+resample: 2
 sampling_rate : 16000

 encoder_decoder:
  depth: 5
-  initial_output_channels: 48
+  initial_output_channels: 32
  kernel_size: 8
  stride: 1
  growth_factor: 2