{
  "student": "/home/henry/Documents/PythonProjects/variable-reap/outputs/pruned/glean-0125inst-math-keep50",
  "teacher": "allenai/OLMoE-1B-7B-0125-Instruct",
  "training_mode": "off-policy",
  "kl_direction": "forward",
  "dataset": "allenai/RLVR-MATH",
  "dataset_sources": null,
  "max_difficulty": null,
  "trajectories": "outputs/teacher_trajectories/dolci_math_curated.jsonl",
  "trajectory_dataset": "allenai/Dolci-Instruct-RL",
  "off_policy_frames": "chat",
  "off_policy_max_seq_len": 2048,
  "topk_targets": "/home/henry/Documents/PythonProjects/variable-reap/outputs/teacher_trajectories/dolci_math_curated_opd_top128",
  "max_loss_tokens": null,
  "loss_tokens_per_step": 120000,
  "teacher_device": "cuda:0",
  "student_device": "cuda:0",
  "lr": 3e-05,
  "optimizer": "adamw8bit",
  "weight_decay": 0.1,
  "epochs": 3,
  "prompts_per_step": 256,
  "group_size": 1,
  "rollout_batch": 64,
  "micro_batch": 3,
  "max_new_tokens": 256,
  "max_prompt_len": 1024,
  "warmup_steps": 10,
  "max_grad_norm": 1.0,
  "eval_every": 10,
  "gsm8k_every": 0,
  "gsm8k_n": 256,
  "gsm8k_batch": 16,
  "gsm8k_max_new_tokens": 512,
  "gsm8k_frames": "chat",
  "save_every": 50,
  "out_dir": "/home/henry/Documents/PythonProjects/variable-reap/outputs/healed/warmup_headcode_probe",
  "sweep": 50,
  "wandb": false,
  "wandb_project": "glean-heal",
  "wandb_run_name": null,
  "wandb_run_id": null,
  "wandb_resume": null,
  "wandb_mode": "offline",
  "no_wandb_sync": false,
  "debug": false,
  "resume_from": null,
  "start_step": 0,
  "no_grad_checkpointing": false,
  "seed": 1224,
  "no_teacher_overlap": false,
  "sync_checkpoints": false,
  "rollout_engine": "hf",
  "vllm_gpu": null,
  "vllm_port": 8377,
  "vllm_refresh_every": 5,
  "vllm_serve_bin": "vllm-plugin/.venv/bin/python",
  "vllm_gpu_mem_util": 0.85,
  "vllm_refresh_mode": "reload",
  "vllm_live_dir": null,
  "resolved_kl_direction": "forward"
}