File size: 159 Bytes
bc87248
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
{
  "architecture_name": "mitra",
  "config": {
    "dim": 512,
    "dim_output": 10,
    "n_layers": 12,
    "n_heads": 4,
    "task": "CLASSIFICATION"
  }
}