NMT-LaVi / config /prototype.json
hieungo1410's picture
'add'
8cb4f3b
raw
history blame
674 Bytes
{
"train_src_data": "/workspace/khoai23/opennmt/data/iwslt_en_vi/train.en",
"train_trg_data": "/workspace/khoai23/opennmt/data/iwslt_en_vi/train.vi",
"valid_src_data": "/workspace/khoai23/opennmt/data/iwslt_en_vi/tst2013.en",
"valid_trg_data": "/workspace/khoai23/opennmt/data/iwslt_en_vi/tst2013.vi",
"src_lang": "en",
"trg_lang": "en",
"max_strlen": 160,
"batchsize": 1500,
"device": "cpu",
"d_model": 512,
"n_layers": 6,
"heads": 8,
"dropout": 0.1,
"lr": 0.0001,
"epochs": 30,
"printevery": 200,
"k": 5,
"n_warmup_steps": 4000,
"beta1": 0.9,
"beta2": 0.98,
"eps": 1e-09,
"label_smoothing": 0.1,
"save_checkpoint_epochs": 5
}