Working on padding

Configuration/config.ini

@@ -5,16 +5,17 @@ transformers = 4.16.2
 symbols_vocab_size=26
 atom_vocab_size=18
 max_len_sentence=290
-max_atoms_in_sentence=874
+max_atoms_in_sentence=875
 max_atoms_in_one_type=324
 
 [MODEL_ENCODER]
 dim_encoder = 768
 
 [MODEL_LINKER]
-nhead=4
+nhead=16
 dim_emb_atom = 256
-num_layers=2
+dim_feedforward_transformer = 512
+num_layers=3
 dim_cat_inter=512
 dim_cat_out=256
 dim_intermediate_FFN=128
@@ -23,7 +24,7 @@ dropout=0.1
 sinkhorn_iters=5
 
 [MODEL_TRAINING]
-batch_size=32
-epoch=25
+batch_size=16
+epoch=30
 seed_val=42
 learning_rate=2e-3
\ No newline at end of file