Working on padding

Configuration/config.ini

@@ -4,7 +4,7 @@ transformers = 4.16.2
 [DATASET_PARAMS]
 symbols_vocab_size=26
 atom_vocab_size=18
-max_len_sentence=83
+max_len_sentence=290
 max_atoms_in_sentence=875
 max_atoms_in_one_type=324
 
@@ -12,7 +12,7 @@ max_atoms_in_one_type=324
 dim_encoder = 768
 
 [MODEL_LINKER]
-nhead=16
+nhead=8
 dim_emb_atom = 256
 dim_feedforward_transformer = 768
 num_layers=3
@@ -25,6 +25,6 @@ sinkhorn_iters=5
 
 [MODEL_TRAINING]
 batch_size=32
-epoch=25
+epoch=30
 seed_val=42
 learning_rate=2e-3
\ No newline at end of file