config.ini

[VERSION]
transformers = 4.16.2

[DATASET_PARAMS]
symbols_vocab_size=26
atom_vocab_size=18
max_len_sentence=290
max_atoms_in_sentence=875
max_atoms_in_one_type=324

[MODEL_ENCODER]
dim_encoder = 768

[MODEL_LINKER]
nhead=8
dim_emb_atom = 256
dim_feedforward_transformer = 768
num_layers=2
dim_cat_inter=512
dim_cat_out=256
dim_intermediate_FFN=128
dim_pre_sinkhorn_transfo=64
dropout=0.1
sinkhorn_iters=5

[MODEL_TRAINING]
batch_size=32
epoch=25
seed_val=42
learning_rate=2e-3