CardioEmbed-BGE-M3 / training_config.json

Cardiology embedding model (separation: 0.209)

196fb7a verified 12 months ago

330 Bytes

{
  "name": "bge-m3",
  "model_id": "BAAI/bge-m3",
  "batch_size": 64,
  "gradient_accumulation_steps": 1,
  "learning_rate": 2e-05,
  "epochs": 3,
  "max_seq_length": 512,
  "lora_r": 16,
  "lora_alpha": 32,
  "use_8bit": false,
  "target_modules": [
    "query",
    "key",
    "value"
  ],
  "output_dir": "bge_m3_cardiology"
}

	{
	"name": "bge-m3",
	"model_id": "BAAI/bge-m3",
	"batch_size": 64,
	"gradient_accumulation_steps": 1,
	"learning_rate": 2e-05,
	"epochs": 3,
	"max_seq_length": 512,
	"lora_r": 16,
	"lora_alpha": 32,
	"use_8bit": false,
	"target_modules": [
	"query",
	"key",
	"value"
	],
	"output_dir": "bge_m3_cardiology"
	}