laya-vulnerability-groups / rl_agent_config.json
peter2000's picture
Laya fine-tuned on GIZ vulnerability data: macro-F1 0.654 (zero-shot 0.108)
ac55aa8 verified
Raw History Blame Contribute Delete
788 Bytes
{
"encoder": "answerdotai/ModernBERT-large",
"head_layers": 2,
"max_len": 512,
"head_max_len": 192,
"max_prefixes": 6,
"act_costs": {
"escalate": 0.5
},
"cost_wrong_act": 3.0,
"amp_dtype": "bf16",
"model_name": "laya-vulnerability-groups",
"temperature": [
1.2,
1.2,
4.928757667541504
],
"temperature_by_options": {
"choice:3-5": 1.7601518630981445,
"choice:6-10": 1.0000158548355103,
"score:3-5": 1.2514300346374512,
"noul:2": 4.928757667541504,
"choice:11+": 0.10058280825614929,
"choice:2": 1.9063563346862793
},
"training": {
"updates": 7313,
"epochs_completed": 1,
"hours": 1.96,
"world_size": 1,
"fine_tuned_from_checkpoint": true
},
"gradient_checkpointing": true,
"fine_tuned": true
}