diff --git a/cli/train_config/config.yaml b/cli/train_config/config.yaml index 61551bd..6b5d98e 100644 --- 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 diff --git a/cli/train_config/model/Demucs.yaml b/cli/train_config/model/Demucs.yaml index 27603dc..1006e71 100644 --- 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