From d6fe0c52c6f0ba8433df1e4e56cac8ef79e820cd Mon Sep 17 00:00:00 2001 From: cfrancesco Date: Thu, 20 May 2021 12:10:32 +0200 Subject: [PATCH] Swapping dimensions is configurable. --- config/aligner_config.yaml | 2 +- config/session_paths_BB.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/config/aligner_config.yaml b/config/aligner_config.yaml index 6534c2a..67eba68 100644 --- a/config/aligner_config.yaml +++ b/config/aligner_config.yaml @@ -1,7 +1,7 @@ # ARCHITECTURE decoder_model_dimension: 256 encoder_model_dimension: 256 -decoder_num_heads: [4, 4, 4, 1] # the length of this defines the number of layers +decoder_num_heads: [4, 4, 4, 4, 1] # the length of this defines the number of layers encoder_num_heads: [4, 4, 4, 4] # the length of this defines the number of layers encoder_feed_forward_dimension: 512 decoder_feed_forward_dimension: 512 diff --git a/config/session_paths_BB.yaml b/config/session_paths_BB.yaml index 89f2799..da458e6 100644 --- a/config/session_paths_BB.yaml +++ b/config/session_paths_BB.yaml @@ -8,5 +8,5 @@ aligner_config: 'config/aligner_config.yaml' tts_config: 'config/tts_config.yaml' tts_config_name: tts_swap_conv_dim -aligner_config_name: aligner_standard +aligner_config_name: aligner_bigger_LN data_name: ljspeech # raw data naming for default data reader (select function from data/metadata_readers.py) \ No newline at end of file