bust / config.json
quebeccyb's picture
Initial release: BuST weights, config, model code, README
75b7c69 verified
raw
history blame contribute delete
422 Bytes
{
"model_type": "bust_v2",
"architectures": ["BuSTv2"],
"task": "text-pair-classification",
"vocab_size": 30003,
"hid_dim": 256,
"n_layers": 3,
"n_heads": 8,
"pf_dim": 512,
"dropout": 0.25,
"max_length": 1024,
"num_classes": 2,
"pad_token_id": 0,
"tokenizer_name": "usmiva/bert-web-bg-cased",
"id2label": {
"0": "human",
"1": "ai"
},
"label2id": {
"human": 0,
"ai": 1
}
}