config.json

{
  "agg_arch": "coeff_mlp",
  "agg_gate_input": "TBD",
  "agg_weight_freeze": "unfrozen",
  "dropout_prob": 0.2,
  "head": "mlp",
  "hidden_dims": [
    512,
    384
  ],
  "inference_mode": true,
  "input_dim": 512,
  "num_classes_per_task": 1,
  "num_tasks": 617,
  "softmax": false,
  "task_type": "classification",
  "use_norm": false
}