En2Kab / config.yaml
Sifal's picture
Upload config.yaml
1126f36
raw
history blame
237 Bytes
eval_batch_size: 256
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