config.json

{
  "backbone": "RN50",
  "d_model": 2048,
  "max_len": 41,
  "encoder_layers": 3,
  "decoder_layers": 1,
  "num_heads": 8,
  "h_dim": 512,
  "a_dim": 2048,
  "dropout": 0.1,
  "learnable": false,
  "fine_tune": true,
  "tie_embeddings": true,
  "prenorm": false,
  "s-transformers": "sentence-transformers/msmarco-distilbert-cos-v5",
  "s-threshold": 1.0,
  "fna": true
}