vdr-msmarco / config.json
jzhoubu's picture
Upload Retriever
0b423c2 verified
Raw
History Blame Contribute Delete
709 Bytes
{
"_name_or_path": "/export/data/jzhoubu/workspace/vsearch/experiments/0112-msmarco-e20-64x4-neg1-lccpu28/train/vdr_20",
"architectures": [
"Retriever"
],
"device": null,
"encoder_p": {
"max_len": 256,
"model_id": "bert-base-uncased",
"norm": false,
"pooling": "max",
"pooling_topk": null,
"shift_vocab_num": 999,
"topk": 768,
"type": "vdr"
},
"encoder_q": {
"max_len": 128,
"model_id": "bert-base-uncased",
"norm": false,
"pooling": "max",
"pooling_topk": null,
"shift_vocab_num": 999,
"topk": 768,
"type": "vdr"
},
"shared_encoder": false,
"torch_dtype": "float32",
"transformers_version": "4.34.0",
"type": null
}