| { | |
| "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 | |
| } |