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