Stevenf232's picture
Upload trained model from Colab
9602540 verified
{
"architectures": [
"HypencoderDualEncoder"
],
"dtype": "float32",
"loss_kwargs": [
{
"use_in_batch_negatives": true
}
],
"loss_type": [
"cross_entropy"
],
"pad_token_id": 0,
"passage_encoder_kwargs": {
"freeze_transformer": true,
"model_name_or_path": "cambridgeltl/SapBERT-from-PubMedBERT-fulltext",
"pooling_type": "cls"
},
"passage_encoder_type": "",
"query_encoder_kwargs": {
"base_encoder_output_dim": 768,
"converter_kwargs": {
"activation_type": "relu",
"do_residual_on_last": false,
"vector_dimensions": [
768,
768,
768,
1
]
},
"embedding_representation": null,
"freeze_transformer": true,
"model_name_or_path": "cambridgeltl/SapBERT-from-PubMedBERT-fulltext"
},
"query_encoder_type": "",
"shared_encoder": true,
"transformers_version": "4.57.3"
}