Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

reduce params

This commit is contained in:
shahules786 2022-09-30 10:55:39 +05:30
parent ba271f8a2a
commit 9170666592
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cli/train_config/config.yaml
View File

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

4
cli/train_config/model/Demucs.yaml
View File

@ -1,11 +1,11 @@
_target_: enhancer.models.demucs.Demucs _target_: enhancer.models.demucs.Demucs
num_channels: 1 num_channels: 1
resample: 4 resample: 2
sampling_rate : 16000 sampling_rate : 16000
encoder_decoder: encoder_decoder:
depth: 5 depth: 5
initial_output_channels: 48 initial_output_channels: 32
kernel_size: 8 kernel_size: 8
stride: 1 stride: 1
growth_factor: 2 growth_factor: 2