Add qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0 step200 LoRA adapter and run config

#4
by Davidk1 - opened
results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/checkpoints/global_step_200/actor/lora_adapter/adapter_config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "task_type": "CAUSAL_LM",
3
+ "peft_type": "LORA",
4
+ "auto_mapping": null,
5
+ "peft_version": "0.19.1",
6
+ "base_model_name_or_path": "/home/yuxinr/.cache/huggingface/hub/models--Qwen--Qwen2.5-1.5B-Instruct/snapshots/989aa7980e4cf806f80c7fef2b1adb7bc71aa306",
7
+ "revision": null,
8
+ "inference_mode": false,
9
+ "r": 64,
10
+ "target_modules": [
11
+ "down_proj",
12
+ "gate_proj",
13
+ "k_proj",
14
+ "o_proj",
15
+ "q_proj",
16
+ "up_proj",
17
+ "v_proj"
18
+ ],
19
+ "exclude_modules": null,
20
+ "lora_alpha": 64,
21
+ "lora_dropout": 0.0,
22
+ "fan_in_fan_out": false,
23
+ "bias": "none",
24
+ "use_rslora": false,
25
+ "modules_to_save": null,
26
+ "init_lora_weights": true,
27
+ "layers_to_transform": null,
28
+ "layers_pattern": null,
29
+ "rank_pattern": {},
30
+ "alpha_pattern": {},
31
+ "megatron_config": null,
32
+ "megatron_core": "megatron.core",
33
+ "trainable_token_indices": null,
34
+ "loftq_config": {},
35
+ "eva_config": null,
36
+ "corda_config": null,
37
+ "lora_ga_config": null,
38
+ "use_dora": false,
39
+ "alora_invocation_tokens": null,
40
+ "use_qalora": false,
41
+ "qalora_group_size": 16,
42
+ "layer_replication": null,
43
+ "runtime_config": {
44
+ "ephemeral_gpu_offload": false
45
+ },
46
+ "lora_bias": false,
47
+ "target_parameters": null,
48
+ "use_bdlora": null,
49
+ "arrow_config": null,
50
+ "ensure_weight_tying": false
51
+ }
results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/checkpoints/global_step_200/actor/lora_adapter/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89aa2560d7d6dbe211e89031626edaa838c8310d5e0a915115de17713beff3a0
3
+ size 295488904
results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/tensorboard/events.out.tfevents.1790134325.train8.3168242.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f479579cd00b1c75908cbdabc26b342fb06b9b2f47b99cafde40f5369c10fb06
3
+ size 38386
results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/tensorboard/events.out.tfevents.1790146713.train8.3292090.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a517c55f5e7af8a117e2816200c2301ef0d85bfadc757b7b8fd85b42d6f20e61
3
+ size 1898681
results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/run_config.json ADDED
@@ -0,0 +1,276 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_model": "989aa7980e4cf806f80c7fef2b1adb7bc71aa306",
3
+ "environment": "webshop",
4
+ "experiment_name": "qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0",
5
+ "implementation_sha256": {
6
+ "scripts/train/clean/run_webshop_1expert_1k_tmux.sh": "505c382630dd089246d8cf7f2101254c932f762338f86291e52118c7eb0d7785",
7
+ "scripts/train/robust/gaussian/run_training.sh": "3c6e237d1a6d27f01d8aef836ae591f1644c4806abefbef27ed453cdfa0c0c8e",
8
+ "verl/models/adaptive_ffn_channel_noise.py": "9b5865cbb6142e06a97f20677369dfe50a885641882231f241b1d48f115f9249",
9
+ "verl/trainer/config/ppo_trainer.yaml": "f0b5654b0bbdf585bad56259942301121d5d694d4bf94aca3f459dd3a6d292e6",
10
+ "verl/workers/actor/dp_actor.py": "fbaa9cb65975acf52404db4443119118419804098fc6d1e18c7b9417e82ea1b8"
11
+ },
12
+ "model_path": "/home/yuxinr/.cache/huggingface/hub/models--Qwen--Qwen2.5-1.5B-Instruct/snapshots/989aa7980e4cf806f80c7fef2b1adb7bc71aa306",
13
+ "perturbation": {
14
+ "actor_weights_perturbed": false,
15
+ "channel_ema_symbol": "lambda_EMA",
16
+ "checkpoint_state_file": "adaptive_ffn_noise.pt",
17
+ "distribution": "multiplicative_normal_scale_1_plus_channel_std_times_epsilon",
18
+ "empty_response_sensitivity_contribution": 0.0,
19
+ "extra_actor_backward_passes_per_rollout_batch": 1,
20
+ "extra_environment_rollouts_per_step": 0,
21
+ "extra_forward_backward_beyond_existing_sensitivity_prepass": 0,
22
+ "family": "stable_gaussian_noise",
23
+ "global_scale_estimator": "squared_L2_norm_of_full_batch_mean_PPO_parameter_gradient_from_sensitivity_prepass",
24
+ "global_scale_used_for": "next_global_step_noise_standard_deviation_only",
25
+ "global_std_cap_after_warmup": 0.4,
26
+ "global_std_cap_multiplier": 2.0,
27
+ "global_std_formula": "sigma_base * warmup * min(updated_policy_full_batch_mean_gradient_squared, clip_multiplier)",
28
+ "global_weight_clipping": true,
29
+ "global_weight_ema": false,
30
+ "gradient_source": "clipped_PPO_policy_objective_only_with_GiGPO_advantages",
31
+ "initial_variance_allocation": "isotropic_before_first_sensitivity_estimate",
32
+ "outer_gradient": "PPO_loss_of_perturbed_policy_on_its_noisy_rollout_against_noisy_vllm_behavior_log_probs_with_identical_channel_scale",
33
+ "rollout_fixed_scale_path_at_engine_initialization": "/home/yuxinr/yuanzhe/RobustLLMAgent/results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/adaptive_rollout_clean_scale.pt",
34
+ "rollout_perturbation": true,
35
+ "rollout_scale_path": "/home/yuxinr/yuanzhe/RobustLLMAgent/results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/adaptive_rollout_scale.pt",
36
+ "rollout_sigma": 0.2,
37
+ "rollout_sigma_schedule": "warmup_times_base_sigma_times_clipped_post_update_gradient_squared",
38
+ "sampling_mode": "step",
39
+ "scope": "one_per_layer_channel_vector_shared_by_all_rollouts_and_actor_mini_batches_in_one_global_step",
40
+ "sensitivity_estimator": "mean_over_rollout_rows_of_squared_channel_gain_gradient_of_sequence_mean_policy_loss",
41
+ "sensitivity_is_fisher": false,
42
+ "sensitivity_normalization": "per_layer_mean_before_EMA",
43
+ "sensitivity_prepass": "updated_noisy_policy_over_entire_existing_rollout_batch_after_all_optimizer_updates",
44
+ "sensitivity_used_for": "next_global_step_variance_only; current_rollout_and_PPO_scale_remains_frozen",
45
+ "settings": {
46
+ "allow_schedule_override": false,
47
+ "alpha": 0.0,
48
+ "damping": 0.01,
49
+ "ema_decay": 0.9,
50
+ "enabled": true,
51
+ "global_scale_clip_multiplier": 2.0,
52
+ "global_scale_fixed_multiplier": -1.0,
53
+ "global_scale_floor": 0.0,
54
+ "global_scale_mode": "prepass_grad_squared",
55
+ "initial_global_scale": 1.0,
56
+ "max_variance_ratio": 4.0,
57
+ "min_variance_ratio": 0.25,
58
+ "seed": 0,
59
+ "sigma": 0.2,
60
+ "statistics_timing": "post_update",
61
+ "warmup_steps": 20
62
+ },
63
+ "statistics_gradient_clipped": false,
64
+ "statistics_timing": "post_update",
65
+ "stop_gradient": [
66
+ "sensitivity",
67
+ "EMA",
68
+ "variance",
69
+ "sampled_channel_noise"
70
+ ],
71
+ "type": "adaptive_gaussian_ffn_channel_noise",
72
+ "update_mode": "lagged_rollout_adaptive",
73
+ "validation_clean_scale_path": "/home/yuxinr/yuanzhe/RobustLLMAgent/results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/adaptive_rollout_clean_scale.pt",
74
+ "validation_perturbation": false,
75
+ "variance_allocation": "inverse_damped_EMA_sensitivity_power_alpha_with_bounded_unit_layer_mean_variance_ratios",
76
+ "variance_projection": "scalar_rescale_with_bounds_and_exact_layer_mean_ratio_1",
77
+ "variance_weight_formula": "(layer_normalized_EMA_sensitivity+damping)^(-alpha)",
78
+ "warmup_formula": "sigma*min((completed_adaptive_updates+1)/warmup_steps,1); full_sigma_when_warmup_steps_is_0"
79
+ },
80
+ "result_category": "robust_training",
81
+ "resume_history": [
82
+ {
83
+ "architecture": "x86_64",
84
+ "dataloader_num_workers": 0,
85
+ "expected_torch": "2.8.0+cu128",
86
+ "free_cache_engine": false,
87
+ "git_commit_at_launch": "6b2abd6d16a4bdfc0987712e01a603a9fd50e324",
88
+ "git_worktree_dirty_at_launch": true,
89
+ "host": "train8",
90
+ "launcher_log": "/home/yuxinr/yuanzhe/RobustLLMAgent/results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0.log",
91
+ "launcher_pid": 3126931,
92
+ "plot_tmux_session": "ws_stable_v9_s200_seed0_plots",
93
+ "project_python": "/home/yuxinr/miniforge3/envs/verl-agent/bin/python",
94
+ "python_no_user_site": true,
95
+ "ray_override_resources": null,
96
+ "ray_temp_dir": "/tmp/ray_wsstablev9200",
97
+ "requested_start_mode": "fresh",
98
+ "resume_checkpoint": null,
99
+ "slurm_cpus_per_task": null,
100
+ "slurm_job_id": null,
101
+ "slurm_mem_per_node": null,
102
+ "tmux_session": "ws_stable_v9_s200_seed0",
103
+ "tmux_socket": null,
104
+ "torch": "2.8.0+cu128",
105
+ "vllm": "0.11.0",
106
+ "vllm_enable_prefix_caching": false,
107
+ "vllm_enforce_eager": false,
108
+ "vllm_load_format": "dummy_dtensor"
109
+ },
110
+ {
111
+ "architecture": "x86_64",
112
+ "dataloader_num_workers": 0,
113
+ "expected_torch": "2.8.0+cu128",
114
+ "free_cache_engine": false,
115
+ "git_commit_at_launch": "6b2abd6d16a4bdfc0987712e01a603a9fd50e324",
116
+ "git_worktree_dirty_at_launch": true,
117
+ "host": "train8",
118
+ "launcher_log": "/home/yuxinr/yuanzhe/RobustLLMAgent/results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0.log",
119
+ "launcher_pid": 3128191,
120
+ "plot_tmux_session": "ws_stable_v9_s200_seed0_plots",
121
+ "project_python": "/home/yuxinr/miniforge3/envs/verl-agent/bin/python",
122
+ "python_no_user_site": true,
123
+ "ray_override_resources": null,
124
+ "ray_temp_dir": "/tmp/ray_wsstablev9200",
125
+ "requested_start_mode": "auto",
126
+ "resume_checkpoint": null,
127
+ "slurm_cpus_per_task": null,
128
+ "slurm_job_id": null,
129
+ "slurm_mem_per_node": null,
130
+ "tmux_session": "ws_stable_v9_s200_seed0",
131
+ "tmux_socket": null,
132
+ "torch": "2.8.0+cu128",
133
+ "vllm": "0.11.0",
134
+ "vllm_enable_prefix_caching": false,
135
+ "vllm_enforce_eager": false,
136
+ "vllm_load_format": "dummy_dtensor"
137
+ },
138
+ {
139
+ "architecture": "x86_64",
140
+ "dataloader_num_workers": 0,
141
+ "expected_torch": "2.8.0+cu128",
142
+ "free_cache_engine": false,
143
+ "git_commit_at_launch": "6b2abd6d16a4bdfc0987712e01a603a9fd50e324",
144
+ "git_worktree_dirty_at_launch": true,
145
+ "host": "train8",
146
+ "launcher_log": "/home/yuxinr/yuanzhe/RobustLLMAgent/results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0.log",
147
+ "launcher_pid": 3145217,
148
+ "plot_tmux_session": "ws_stable_v9_s200_seed0_plots",
149
+ "project_python": "/home/yuxinr/miniforge3/envs/verl-agent/bin/python",
150
+ "python_no_user_site": true,
151
+ "ray_override_resources": null,
152
+ "ray_temp_dir": "/tmp/ray_wsstablev9200",
153
+ "requested_start_mode": "auto",
154
+ "resume_checkpoint": null,
155
+ "slurm_cpus_per_task": null,
156
+ "slurm_job_id": null,
157
+ "slurm_mem_per_node": null,
158
+ "tmux_session": "ws_stable_v9_s200_seed0",
159
+ "tmux_socket": null,
160
+ "torch": "2.8.0+cu128",
161
+ "vllm": "0.11.0",
162
+ "vllm_enable_prefix_caching": false,
163
+ "vllm_enforce_eager": false,
164
+ "vllm_load_format": "dummy_dtensor"
165
+ },
166
+ {
167
+ "architecture": "x86_64",
168
+ "dataloader_num_workers": 0,
169
+ "expected_torch": "2.8.0+cu128",
170
+ "free_cache_engine": false,
171
+ "git_commit_at_launch": "6b2abd6d16a4bdfc0987712e01a603a9fd50e324",
172
+ "git_worktree_dirty_at_launch": true,
173
+ "host": "train8",
174
+ "launcher_log": "/home/yuxinr/yuanzhe/RobustLLMAgent/results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0.log",
175
+ "launcher_pid": 3154938,
176
+ "plot_tmux_session": "ws_stable_v9_s200_seed0_plots",
177
+ "project_python": "/home/yuxinr/miniforge3/envs/verl-agent/bin/python",
178
+ "python_no_user_site": true,
179
+ "ray_override_resources": null,
180
+ "ray_temp_dir": "/tmp/ray_wsstablev9200",
181
+ "requested_start_mode": "auto",
182
+ "resume_checkpoint": null,
183
+ "slurm_cpus_per_task": null,
184
+ "slurm_job_id": null,
185
+ "slurm_mem_per_node": null,
186
+ "tmux_session": "ws_stable_v9_s200_seed0",
187
+ "tmux_socket": null,
188
+ "torch": "2.8.0+cu128",
189
+ "vllm": "0.11.0",
190
+ "vllm_enable_prefix_caching": false,
191
+ "vllm_enforce_eager": false,
192
+ "vllm_load_format": "dummy_dtensor"
193
+ }
194
+ ],
195
+ "runtime": {
196
+ "architecture": "x86_64",
197
+ "dataloader_num_workers": 0,
198
+ "expected_torch": "2.8.0+cu128",
199
+ "free_cache_engine": false,
200
+ "git_commit_at_launch": "6b2abd6d16a4bdfc0987712e01a603a9fd50e324",
201
+ "git_worktree_dirty_at_launch": true,
202
+ "host": "train8",
203
+ "launcher_log": "/home/yuxinr/yuanzhe/RobustLLMAgent/results/webshop/robust_training/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0/logs/qwen2_5_1_5_stable_gaussian_noise_v9_step_200_seed0.log",
204
+ "launcher_pid": 3278823,
205
+ "plot_tmux_session": "ws_stable_v9_s200_seed0_plots",
206
+ "project_python": "/home/yuxinr/miniforge3/envs/verl-agent/bin/python",
207
+ "python_no_user_site": true,
208
+ "ray_override_resources": null,
209
+ "ray_temp_dir": "/tmp/ray_wsstablev9200",
210
+ "requested_start_mode": "auto",
211
+ "resume_checkpoint": null,
212
+ "slurm_cpus_per_task": null,
213
+ "slurm_job_id": null,
214
+ "slurm_mem_per_node": null,
215
+ "tmux_session": "ws_stable_v9_s200_seed0",
216
+ "tmux_socket": null,
217
+ "torch": "2.8.0+cu128",
218
+ "vllm": "0.11.0",
219
+ "vllm_enable_prefix_caching": false,
220
+ "vllm_enforce_eager": false,
221
+ "vllm_load_format": "dummy_dtensor"
222
+ },
223
+ "source_checkpoint": null,
224
+ "status": "active",
225
+ "training": {
226
+ "actor_micro_batch_size_per_gpu": 4,
227
+ "algorithm": "GiGPO",
228
+ "cleanup_non_interval_checkpoints_on_completion": true,
229
+ "clip_ratio_high": 0.2,
230
+ "clip_ratio_low": 0.2,
231
+ "completed_checkpoint_keep_interval": 100,
232
+ "cuda_visible_devices": "0,1",
233
+ "entropy_coeff": 0.001,
234
+ "gamma": 0.95,
235
+ "history_length": 2,
236
+ "invalid_action_penalty_coef": 0.1,
237
+ "learning_rate": 3e-06,
238
+ "lora_alpha": 64,
239
+ "lora_rank": 64,
240
+ "max_environment_steps": 15,
241
+ "max_prompt_length": 4096,
242
+ "max_response_length": 512,
243
+ "n_gpus_per_node": 2,
244
+ "old_log_probs_source": "noisy_vllm_rollout_with_same_channel_scale_as_PPO",
245
+ "optimizer_grad_clip": 1.0,
246
+ "ppo_epochs": 1,
247
+ "ppo_mini_batch_size": 128,
248
+ "resume_mode": "auto",
249
+ "robust_update_count": 200,
250
+ "rollout_gpu_memory_utilization": 0.3,
251
+ "rollout_n": 8,
252
+ "rollout_temperature": 1.0,
253
+ "rollout_tensor_model_parallel_size": 1,
254
+ "rollout_top_p": 1.0,
255
+ "save_frequency_steps": 5,
256
+ "seed": 0,
257
+ "start_mode": "fresh",
258
+ "target_global_step": 200,
259
+ "thinking_mode": "off",
260
+ "thinking_mode_launcher": "nothinking",
261
+ "train_batch_size": 16,
262
+ "train_sessions_per_worker": 8,
263
+ "use_kl_loss": false,
264
+ "val_sessions_per_worker": 8
265
+ },
266
+ "validation": {
267
+ "channel_scale": "explicit_all_ones_then_restore_training_scale",
268
+ "do_sample": true,
269
+ "frequency_steps": 20,
270
+ "official_evaluation_goals": "0..499 (separate post-training evaluation)",
271
+ "perturbation_enabled": false,
272
+ "sampled_goal_count": 128,
273
+ "step_0_enabled": true,
274
+ "temperature": 0.4
275
+ }
276
+ }