MR0b0t's picture
Add config.json (architecture metadata, enables Hub download counting); card: download snippet
b4525a4 verified
Raw History Blame Contribute Delete
560 Bytes
{
"model_type": "encoder-decoder-transformer",
"_note": "Architecture metadata. Also the file the Hugging Face Hub counts downloads on for repos without a registered library.",
"d_model": 256,
"num_encoder_layers": 4,
"num_decoder_layers": 4,
"num_heads": 8,
"d_ff": 512,
"dropout": 0.1,
"max_seq_len": 300,
"src_vocab_size": 14476,
"tgt_vocab_size": 14512,
"src_language": "C++",
"tgt_language": "Python",
"train_data": "XLCoST C++-Python pair_data_tok_full",
"checkpoint": "best_model.pt",
"epoch": 19,
"val_loss": 2.0474
}