Caroline de Pourtalès
--- a/Configuration/config.ini

+ 3

− 3
+++ b/Configuration/config.ini

+ 3

− 3
 @@ -4,7 +4,7 @@ transformers = 4.16.2
 @@ -4,7 +4,7 @@ transformers = 4.16.2
 [DATASET_PARAMS]
 symbols_vocab_size=26
 atom_vocab_size=18
-max_len_sentence=290
+max_len_sentence=83
 max_atoms_in_sentence=875
 max_atoms_in_one_type=324
 @@ -12,10 +12,10 @@ max_atoms_in_one_type=324
 @@ -12,10 +12,10 @@ max_atoms_in_one_type=324
 dim_encoder = 768
 [MODEL_LINKER]
-nhead=8
+nhead=16
 dim_emb_atom = 256
 dim_feedforward_transformer = 768
-num_layers=2
+num_layers=3
 dim_cat_inter=512
 dim_cat_out=256
 dim_intermediate_FFN=128