File size: 334 Bytes
e00b943 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | {
"model_type": "transformer_seq2seq",
"embed_dim": 768,
"vocab_size": 64000,
"num_heads": 4,
"num_layers": 2,
"dim_feedforward": 256,
"dropout": 0.1,
"freeze_decoder_emb": true,
"max_len": 128,
"batch_size": 32,
"source_tokenizer": "bert-base-cased-finetuned-mrpc",
"target_tokenizer": "vinai/phobert-base"
}
|