Swapping dimensions is configurable.

jayinc · May 20, 2021 · d6fe0c5 · d6fe0c5
1 parent 2a23626
commit d6fe0c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)