Add RL job config
Browse files- training_logs/rl_config.json +156 -0
training_logs/rl_config.json
ADDED
|
@@ -0,0 +1,156 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"job_name": "a3-rl-laion_exp_rpt_codenet-python-v2",
|
| 3 |
+
"experiments_dir": "/e/data1/datasets/playground/ot-baf/a3-rl-laion_exp_rpt_codenet-python-v2",
|
| 4 |
+
"cluster_name": "jupiter",
|
| 5 |
+
"skyrl_entrypoint": "examples.terminal_bench.entrypoints.main_tbench",
|
| 6 |
+
"skyrl_hydra_args": [
|
| 7 |
+
"+terminal_bench_config=terminal_bench",
|
| 8 |
+
"trainer.strategy=fsdp2",
|
| 9 |
+
"trainer.algorithm.advantage_estimator=rloo_n",
|
| 10 |
+
"trainer.algorithm.use_kl_loss=false",
|
| 11 |
+
"trainer.algorithm.kl_loss_coef=0.0",
|
| 12 |
+
"trainer.algorithm.eps_clip_low=0.2",
|
| 13 |
+
"trainer.algorithm.eps_clip_high=0.05",
|
| 14 |
+
"trainer.algorithm.loss_reduction=token_mean",
|
| 15 |
+
"trainer.epochs=2",
|
| 16 |
+
"trainer.max_steps=80",
|
| 17 |
+
"trainer.update_epochs_per_batch=1",
|
| 18 |
+
"trainer.train_batch_size=64",
|
| 19 |
+
"trainer.policy_mini_batch_size=64",
|
| 20 |
+
"trainer.eval_batch_size=64",
|
| 21 |
+
"trainer.micro_forward_batch_size_per_gpu=4",
|
| 22 |
+
"trainer.micro_train_batch_size_per_gpu=1",
|
| 23 |
+
"trainer.max_prompt_length=999999",
|
| 24 |
+
"trainer.eval_interval=999999",
|
| 25 |
+
"trainer.eval_before_train=false",
|
| 26 |
+
"trainer.ckpt_interval=2",
|
| 27 |
+
"trainer.resume_mode=latest",
|
| 28 |
+
"trainer.hf_save_interval=5",
|
| 29 |
+
"++trainer.hf_hub_repo_id=laion/a3-rl-laion_exp_rpt_codenet-python-v2",
|
| 30 |
+
"++trainer.hf_hub_private=false",
|
| 31 |
+
"++trainer.hf_hub_revision=main",
|
| 32 |
+
"++trainer.enable_db_registration=false",
|
| 33 |
+
"trainer.project_name=OpenThoughts-Agent",
|
| 34 |
+
"trainer.log_level=INFO",
|
| 35 |
+
"trainer.tracker_commit_each_step=true",
|
| 36 |
+
"trainer.logger=console",
|
| 37 |
+
"trainer.run_name=a3-rl-laion_exp_rpt_codenet-python-v2",
|
| 38 |
+
"trainer.ckpt_path=/e/data1/datasets/playground/ot-baf/a3-rl-laion_exp_rpt_codenet-python-v2/a3-rl-laion_exp_rpt_codenet-python-v2/checkpoints",
|
| 39 |
+
"trainer.export_path=/e/data1/datasets/playground/ot-baf/a3-rl-laion_exp_rpt_codenet-python-v2/a3-rl-laion_exp_rpt_codenet-python-v2/exports",
|
| 40 |
+
"trainer.policy.optimizer_config.lr=8e-6",
|
| 41 |
+
"trainer.policy.optimizer_config.weight_decay=0.0",
|
| 42 |
+
"trainer.policy.optimizer_config.adam_betas=[0.9,0.999]",
|
| 43 |
+
"trainer.policy.optimizer_config.max_grad_norm=0.9",
|
| 44 |
+
"trainer.policy.fsdp_config.cpu_offload=false",
|
| 45 |
+
"trainer.policy.fsdp_config.reshard_after_forward=true",
|
| 46 |
+
"trainer.policy.fsdp_config.fsdp_size=4",
|
| 47 |
+
"trainer.policy.model.path=/e/home/jusers/feuer1/jupiter/.cache/huggingface/hub/models--laion--GLM-4_7-swesmith-sandboxes-with_tests-oracle_verified_120s-maxeps-131k-fixthink/snapshots/0e3bff0c4e51f6b9ec0713b98b9eec36efb91cc6",
|
| 48 |
+
"trainer.ref.fsdp_config.cpu_offload=false",
|
| 49 |
+
"trainer.ref.fsdp_config.reshard_after_forward=true",
|
| 50 |
+
"trainer.ref.fsdp_config.fsdp_size=4",
|
| 51 |
+
"trainer.placement.colocate_all=false",
|
| 52 |
+
"trainer.placement.policy_num_nodes=2",
|
| 53 |
+
"trainer.placement.ref_num_nodes=2",
|
| 54 |
+
"trainer.placement.policy_num_gpus_per_node=4",
|
| 55 |
+
"trainer.placement.ref_num_gpus_per_node=4",
|
| 56 |
+
"trainer.fully_async.max_staleness_steps=16",
|
| 57 |
+
"trainer.fully_async.num_parallel_generation_workers=338",
|
| 58 |
+
"generator.backend=vllm",
|
| 59 |
+
"generator.timeout_multiplier=1.0",
|
| 60 |
+
"generator.model_dtype=bfloat16",
|
| 61 |
+
"generator.inference_engine_tensor_parallel_size=1",
|
| 62 |
+
"generator.num_inference_engines=48",
|
| 63 |
+
"generator.n_samples_per_prompt=8",
|
| 64 |
+
"generator.eval_n_samples_per_prompt=8",
|
| 65 |
+
"generator.gpu_memory_utilization=0.75",
|
| 66 |
+
"generator.max_num_seqs=24",
|
| 67 |
+
"generator.max_num_batched_tokens=65536",
|
| 68 |
+
"generator.enable_prefix_caching=true",
|
| 69 |
+
"generator.enable_chunked_prefill=true",
|
| 70 |
+
"generator.run_engines_locally=true",
|
| 71 |
+
"generator.weight_sync_backend=nccl",
|
| 72 |
+
"generator.async_engine=true",
|
| 73 |
+
"generator.batched=false",
|
| 74 |
+
"generator.enable_http_endpoint=true",
|
| 75 |
+
"generator.enable_ray_prometheus_stats=false",
|
| 76 |
+
"generator.vllm_stats_interval=1",
|
| 77 |
+
"generator.append_eos_token_after_stop_str_in_multi_turn=true",
|
| 78 |
+
"generator.max_turns=999999",
|
| 79 |
+
"generator.sampling_params.max_generate_length=4096",
|
| 80 |
+
"generator.sampling_params.temperature=0.7",
|
| 81 |
+
"generator.sampling_params.top_p=0.95",
|
| 82 |
+
"generator.sampling_params.top_k=20",
|
| 83 |
+
"++generator.engine_init_kwargs.max_model_len=32768",
|
| 84 |
+
"++generator.engine_init_kwargs.custom_chat_template_chat_completion_path=chat_templates/qwen3_thinking_acc.jinja2",
|
| 85 |
+
"++generator.engine_init_kwargs.served_model_name=0e3bff0c4e51f6b9ec0713b98b9eec36efb91cc6",
|
| 86 |
+
"data.train_data=[\"/e/scratch/jureap59/feuer1/tasks/exp_rpt_codenet-python-v2\"]",
|
| 87 |
+
"data.val_data=[]",
|
| 88 |
+
"+terminal_bench_config.trials_dir=/e/data1/datasets/playground/ot-baf/a3-rl-laion_exp_rpt_codenet-python-v2/a3-rl-laion_exp_rpt_codenet-python-v2/trace_jobs",
|
| 89 |
+
"+terminal_bench_config.harbor.name=terminus-2",
|
| 90 |
+
"+terminal_bench_config.harbor.max_episodes=999999",
|
| 91 |
+
"+terminal_bench_config.harbor.enable_summarize=false",
|
| 92 |
+
"+terminal_bench_config.harbor.store_all_messages=true",
|
| 93 |
+
"+terminal_bench_config.harbor.trajectory_config.raw_content=true",
|
| 94 |
+
"+terminal_bench_config.harbor.enable_episode_logging=false",
|
| 95 |
+
"+terminal_bench_config.harbor.record_terminal_session=false",
|
| 96 |
+
"+terminal_bench_config.harbor.enable_pane_logging=false",
|
| 97 |
+
"+terminal_bench_config.harbor.strict_json_parser=true",
|
| 98 |
+
"+terminal_bench_config.harbor.interleaved_thinking=true",
|
| 99 |
+
"+terminal_bench_config.harbor.extra_body.chat_template_kwargs.enable_thinking=true",
|
| 100 |
+
"+terminal_bench_config.harbor.override_timeout_sec=900",
|
| 101 |
+
"+terminal_bench_config.harbor.override_cpus=1",
|
| 102 |
+
"+terminal_bench_config.harbor.override_memory_mb=2048",
|
| 103 |
+
"+terminal_bench_config.harbor.override_storage_mb=2048",
|
| 104 |
+
"+terminal_bench_config.harbor.auto_snapshot=true",
|
| 105 |
+
"+terminal_bench_config.harbor.verifier_override_timeout_sec=120",
|
| 106 |
+
"+terminal_bench_config.harbor.max_retries=3",
|
| 107 |
+
"+terminal_bench_config.harbor.min_wait_sec=60.0",
|
| 108 |
+
"+terminal_bench_config.harbor.max_wait_sec=600.0",
|
| 109 |
+
"+terminal_bench_config.harbor.wait_multiplier=2.0",
|
| 110 |
+
"+terminal_bench_config.harbor.exclude_exceptions=[\"VerifierTimeoutError\",\"VerifierRuntimeError\",\"RewardFileNotFoundError\",\"RewardFileEmptyError\",\"VerifierOutputParseError\"]",
|
| 111 |
+
"+terminal_bench_config.harbor.n_concurrent_trials=675",
|
| 112 |
+
"+terminal_bench_config.harbor.log_level=INFO",
|
| 113 |
+
"+terminal_bench_config.harbor.enable_reward_shaping=false",
|
| 114 |
+
"+terminal_bench_config.harbor.enable_error_classification=true",
|
| 115 |
+
"+terminal_bench_config.harbor.mask_exceptions=[\"DaytonaError\",\"EnvironmentStartTimeoutError\",\"NetworkError\",\"ConnectionError\",\"RewardFileNotFoundError\",\"RewardFileEmptyError\",\"AgentEnvironmentTimeoutError\",\"ContextLengthExceededError\"]",
|
| 116 |
+
"+terminal_bench_config.harbor.default_error_treatment=zero",
|
| 117 |
+
"+terminal_bench_config.harbor.passthrough_exceptions=[\"AgentTimeoutError\"]",
|
| 118 |
+
"+terminal_bench_config.harbor.zero_exceptions=[]",
|
| 119 |
+
"+terminal_bench_config.model_info.max_input_tokens=32000",
|
| 120 |
+
"+terminal_bench_config.model_info.max_output_tokens=4096",
|
| 121 |
+
"+terminal_bench_config.archiving.enabled=false",
|
| 122 |
+
"+terminal_bench_config.trace_upload.enabled=true",
|
| 123 |
+
"+terminal_bench_config.trace_upload.repo_org=DCAgent",
|
| 124 |
+
"+terminal_bench_config.trace_upload.episodes=last",
|
| 125 |
+
"+terminal_bench_config.trace_upload.dataset_type=SFT",
|
| 126 |
+
"+terminal_bench_config.trace_upload.cleanup=true"
|
| 127 |
+
],
|
| 128 |
+
"model_path": "/e/home/jusers/feuer1/jupiter/.cache/huggingface/hub/models--laion--GLM-4_7-swesmith-sandboxes-with_tests-oracle_verified_120s-maxeps-131k-fixthink/snapshots/0e3bff0c4e51f6b9ec0713b98b9eec36efb91cc6",
|
| 129 |
+
"train_data": [
|
| 130 |
+
"/e/scratch/jureap59/feuer1/tasks/exp_rpt_codenet-python-v2"
|
| 131 |
+
],
|
| 132 |
+
"val_data": [],
|
| 133 |
+
"num_nodes": 14,
|
| 134 |
+
"gpus_per_node": 4,
|
| 135 |
+
"cpus_per_node": 288,
|
| 136 |
+
"tensor_parallel_size": 1,
|
| 137 |
+
"ray_port": 6379,
|
| 138 |
+
"master_port": 12345,
|
| 139 |
+
"checkpoints_dir": null,
|
| 140 |
+
"export_path": "/e/data1/datasets/playground/ot-baf/a3-rl-laion_exp_rpt_codenet-python-v2/a3-rl-laion_exp_rpt_codenet-python-v2/exports",
|
| 141 |
+
"needs_ssh_tunnel": true,
|
| 142 |
+
"needs_cuda_detection": false,
|
| 143 |
+
"pinggy_persistent_url": null,
|
| 144 |
+
"pinggy_token": null,
|
| 145 |
+
"agent_name": "terminus-2",
|
| 146 |
+
"harbor_env": "daytona",
|
| 147 |
+
"proxychains_binary": null,
|
| 148 |
+
"container_sif": null,
|
| 149 |
+
"container_binds": [],
|
| 150 |
+
"ray_object_store_gb": 40.0,
|
| 151 |
+
"trace_upload_enabled": true,
|
| 152 |
+
"trace_upload_repo_org": "DCAgent",
|
| 153 |
+
"trace_upload_episodes": "last",
|
| 154 |
+
"trace_upload_dataset_type": "SFT",
|
| 155 |
+
"trace_upload_cleanup": true
|
| 156 |
+
}
|