rl_agent_config.json

{
  "encoder": "dicta-il/neodictabert-bilingual",
  "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-hebrew",
  "temperature": [
    1.0730161666870117,
    1.1290489435195923,
    1.1732172966003418
  ],
  "training": {
    "init": "/content/drive/MyDrive/laya_he_v3/model.safetensors",
    "seed": 1,
    "items": 396508,
    "updates": 6196,
    "epochs_completed": 1,
    "lr_encoder": 5e-06,
    "lr_head": 0.0001,
    "hours": 1.42,
    "calibration_items": {
      "0": 600,
      "1": 600,
      "2": 600
    },
    "avg_loss": 0.3189
  },
  "trust_remote_code": true,
  "gradient_checkpointing": true,
  "max_tokens_per_batch": 4096,
  "fine_tuned": true
}