File size: 481 Bytes
ea86ebc 027f08a ea86ebc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
{
"model_type": "superposition_patch_classifier",
"embed_dim": 256,
"patch_dim": 64,
"n_bootstrap": 2,
"n_geometric": 2,
"n_heads": 4,
"dropout": 0.0,
"grid_shape": [8, 16, 16],
"patch_shape": [2, 4, 4],
"num_patches": 64,
"local_gate_dim": 11,
"structural_gate_dim": 6,
"total_gate_dim": 17,
"num_classes": 27,
"num_gates": 5,
"auto_map": {
"AutoModel": "geometric_model.SuperpositionPatchClassifier"
}
} |