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