File size: 340 Bytes
aec320b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
{
  "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": 12,
  "softmax": false,
  "task_type": "classification",
  "use_norm": false
}