File size: 867 Bytes
15d1a73 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
{
"text_emb_model_config": {
"_target_": "gfmrag.text_emb_models.BaseTextEmbModel",
"text_emb_model_name": "sentence-transformers/all-mpnet-base-v2",
"normalize": false,
"batch_size": 32,
"query_instruct": null,
"passage_instruct": null,
"model_kwargs": null
},
"model_config": {
"_target_": "gfmrag.models.GNNRetriever",
"entity_model": {
"_target_": "gfmrag.ultra.models.QueryNBFNet",
"input_dim": 512,
"hidden_dims": [
512,
512,
512,
512,
512,
512
],
"message_func": "distmult",
"aggregate_func": "sum",
"short_cut": true,
"layer_norm": true
},
"rel_emb_dim": 768
}
}
|