File size: 228 Bytes
8b41d42
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
{
    "model_name": "daedalus_mobile",
    "num_layers": 6,
    "num_heads": 8,
    "hidden_size": 512,
    "dropout": 0.1,
    "lr": 1e-4,
    "batch_size": 32,
    "epochs": 5,
    "max_seq_length": 512,
    "device": "cuda"
}