File size: 178 Bytes
fb6b7f8
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
{
    "d_model": 768,
    "num_heads": 12,
    "num_kv_heads": 6,
    "num_layers": 12,
    "ff_dim": 3072,
    "batch_size": 20,
    "freeze": [0, 1, 2],
    "num_workers": 4
}