train_config.json

{
  "model_name": "sentence-transformers/all-MiniLM-L6-v2",
  "max_length": 32,
  "epochs": 6,
  "batch_size": 16,
  "lr_encoder": 2e-05,
  "lr_heads": 0.0008,
  "weight_decay": 0.01,
  "task_weight_macro": 1.2,
  "task_weight_intent": 1.0,
  "task_weight_context": 1.6,
  "device": "cuda",
  "output_dir": "artifacts",
  "seed": 42
}