{
  "schema_version": 1,
  "run": {
    "name": "unified-multitask-augmented-v3",
    "seed": 42,
    "workspace": "/workspace/patronus-unified-v3"
  },
  "dataset": {
    "name": "unified-multitask-data-v3",
    "repo_id": "patronus-studio/unified-multitask-data-v3",
    "revision": "6733a9fef36f059f0717841a16902e2de4d0d7de",
    "format": "jsonl",
    "files": {
      "train": "data/train.jsonl",
      "validation": "data/val.jsonl",
      "test": "data/test.jsonl"
    },
    "split_values": {
      "train": "train",
      "validation": "val",
      "test": "test"
    }
  },
  "model": {
    "base_model": "jhu-clsp/mmBERT-small",
    "revision": "abc32620dd4f6ab06f5fbe905dc25f310618e09f",
    "max_length": 256
  },
  "training": {
    "epochs": 3,
    "batch_size": 32,
    "gradient_accumulation": 1,
    "learning_rate": 2e-05,
    "weight_decay": 0.01,
    "mixed_precision": "bf16",
    "num_workers": 4
  },
  "evaluation": {
    "primary_metric": "mean_head_f1",
    "minimum_primary_metric": 0.0
  },
  "publish": {
    "sync_name": "unified-multitask-model-augmented-v3",
    "repo_id": "patronus-studio/unified-multitask-model-augmented-v3",
    "single_head_repos": {
      "injection": "patronus-studio/unified-multitask-model-augmented-v3-injection",
      "sensitive_document": "patronus-studio/unified-multitask-model-augmented-v3-sensitive-document",
      "tool_class": "patronus-studio/unified-multitask-model-augmented-v3-tool-class",
      "tool_action": "patronus-studio/unified-multitask-model-augmented-v3-tool-action",
      "tool_tags": "patronus-studio/unified-multitask-model-augmented-v3-tool-tags",
      "routing": "patronus-studio/unified-multitask-model-augmented-v3-routing",
      "threat": "patronus-studio/unified-multitask-model-augmented-v3-threat"
    },
    "private": true
  },
  "resolved": {
    "dataset_dir": "/workspace/patronus-unified-v3/dataset",
    "data_files": {
      "train": "/workspace/patronus-unified-v3/dataset/data/train.jsonl",
      "validation": "/workspace/patronus-unified-v3/dataset/data/val.jsonl",
      "test": "/workspace/patronus-unified-v3/dataset/data/test.jsonl"
    },
    "dataset_report": "/workspace/patronus-unified-v3/dataset_report.json"
  },
  "single_head": "tool_action"
}