config.yaml

ffn_hid_dim: 2048
num_encoder_layers: 5
num_decoder_layers: 5
emb_size: 512
nhead: 2
source_vocab_size: 30524
target_vocab_size: 35008
add_special_tokens: true
return_tensors: pt
src_max_length: 21
tgt_max_length: 22