train_config.json

{
  "model_name": "sentence-transformers/all-MiniLM-L6-v2",
  "max_length": 128,
  "epochs": 20,
  "batch_size": 10,
  "lr_encoder": 2e-05,
  "lr_heads": 0.001,
  "weight_decay": 0.01,
  "task_weight_macro": 1.0,
  "task_weight_intent": 1.0,
  "task_weight_context": 1.0,
  "device": "cuda",
  "output_dir": "artifacts",
  "seed": 42
}