config.ini

[VERSION]
transformers = 4.16.2

[DATASET_PARAMS]
symbols_vocab_size=26
atom_vocab_size=12
max_len_sentence=148
max_symbols_in_sentence=1250

[MODEL_ENCODER]
dim_encoder = 768

[MODEL_DECODER]
dim_encoder = 768
dim_decoder = 8
num_rnn_layers=1
dropout=0.1
teacher_forcing=0.05

[MODEL_LINKER]
nhead=8
dim_feedforward=246
dim_embedding_atoms=8
dim_polarity_transfo=128
layer_norm_eps=1e-5
dropout=0.1
sinkhorn_iters=3

[MODEL_TRAINING]
device=cpu
batch_size=32
epoch=20
seed_val=42
learning_rate=0.005
use_checkpoint_SAVE=0
output_path=Output
use_checkpoint_LOAD=0
input_path=Input
model_to_load=model_check.pt