penfever commited on
Commit
263605d
·
verified ·
1 Parent(s): b52fadb

Add RL job config

Browse files
Files changed (1) hide show
  1. 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
+ }