safe-laya / rl_agent_config.json
ottosulin's picture
safe-laya v5 CUDA fine-tune
7e7dc73 verified
Raw History Blame Contribute Delete
557 Bytes
{
"encoder": "answerdotai/ModernBERT-large",
"head_layers": 2,
"max_len": 1024,
"head_max_len": 256,
"max_prefixes": 6,
"act_costs": {
"escalate": 0.5
},
"cost_wrong_act": 3.0,
"amp_dtype": "bf16",
"model_name": "laya-safe-jeff-v5",
"temperature": [
2.014085531234741,
1.2,
1.2
],
"training": {
"updates": 7313,
"epochs_completed": 1,
"hours": 1.96,
"world_size": 1,
"fine_tuned_from_checkpoint": true
},
"max_tokens_per_batch": 4096,
"gradient_checkpointing": true,
"fine_tuned": true
}