Hugging Face's logo

zaleni
/
MagicBot-VGA-Libero

Safetensors
Model card Files
xet
 Community
zaleni commited on
Commit
b8a24a0
·
verified ·
1 Parent(s): 1c6892e

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. config.json +108 -0
  2. model.safetensors +3 -0
  3. stats.json +321 -0
  4. train_config.json +307 -0
config.json ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "cubev2",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 32
9
+ ]
10
+ }
11
+ },
12
+ "output_features": {
13
+ "action": {
14
+ "type": "ACTION",
15
+ "shape": [
16
+ 32
17
+ ]
18
+ }
19
+ },
20
+ "device": "cuda:0",
21
+ "use_amp": false,
22
+ "push_to_hub": false,
23
+ "repo_id": "lerobot_lab/cubev2",
24
+ "private": null,
25
+ "tags": null,
26
+ "license": null,
27
+ "pretrained_path": "/home/jiangjiahao/data/model/MagicBot-VGA-Base",
28
+ "qwen3_vl_variant": "qwen3_vl_28l",
29
+ "action_expert_variant": "qwen3_28l",
30
+ "qwen3_vl_pretrained_path": "/home/jiangjiahao/data/model/Qwen3-VL-2B-Instruct",
31
+ "dtype": "bfloat16",
32
+ "chunk_size": 10,
33
+ "n_action_steps": 10,
34
+ "max_state_dim": 32,
35
+ "max_action_dim": 32,
36
+ "num_inference_steps": 10,
37
+ "time_sampling_beta_alpha": 1.5,
38
+ "time_sampling_beta_beta": 1.0,
39
+ "time_sampling_scale": 0.999,
40
+ "time_sampling_offset": 0.001,
41
+ "min_period": 0.004,
42
+ "max_period": 4.0,
43
+ "image_resolution": [
44
+ 224,
45
+ 224
46
+ ],
47
+ "empty_cameras": 0,
48
+ "normalization_mapping": {
49
+ "VISUAL": "IDENTITY",
50
+ "STATE": "IDENTITY",
51
+ "ACTION": "IDENTITY"
52
+ },
53
+ "gradient_checkpointing": false,
54
+ "compile_model": false,
55
+ "compile_mode": "max-autotune",
56
+ "optimizer_lr": 5e-05,
57
+ "optimizer_betas": [
58
+ 0.9,
59
+ 0.95
60
+ ],
61
+ "optimizer_eps": 1e-08,
62
+ "optimizer_weight_decay": 0.01,
63
+ "optimizer_grad_clip_norm": 1.0,
64
+ "scheduler_warmup_steps": 2000,
65
+ "scheduler_decay_steps": 80000,
66
+ "scheduler_decay_lr": 5e-06,
67
+ "tokenizer_max_length": 48,
68
+ "freeze_vision_encoder": false,
69
+ "train_expert_only": false,
70
+ "train_vlm_only": false,
71
+ "scale_factor": 8,
72
+ "lambda_gen": 0.01,
73
+ "cosmos_tokenizer_path_or_name": "/home/jiangjiahao/data/model/Cosmos-Tokenizer-CI8x8",
74
+ "enable_3d_queries": true,
75
+ "num_3d_query_tokens": 432,
76
+ "da3_alignment_mode": "query_decoder",
77
+ "da3_query_resampler_layers": 1,
78
+ "da3_query_resampler_ff_mult": 1,
79
+ "query_layer_indices": [
80
+ 13,
81
+ 19,
82
+ 23,
83
+ 27
84
+ ],
85
+ "da3_variant": "auto",
86
+ "da3_teacher_layers": [
87
+ 11,
88
+ 15,
89
+ 19,
90
+ 23
91
+ ],
92
+ "da3_query_dim": 2048,
93
+ "da3_tokens_per_view": 1296,
94
+ "da3_num_views": 3,
95
+ "lambda_3d": 0.01,
96
+ "da3_model_path_or_name": "/home/jiangjiahao/data/model/DA3-LARGE-1.1",
97
+ "da3_model_name": null,
98
+ "da3_code_root": null,
99
+ "da3_teacher_process_res": 504,
100
+ "da3_layer_weights": [
101
+ 1.0,
102
+ 1.2,
103
+ 1.4,
104
+ 1.6
105
+ ],
106
+ "future_query_init_std": 0.02,
107
+ "log_da3_teacher_timing": true
108
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2b15dd0537a3998415630253d21a43f0b74d0e66d84ccaa8bd743742cdbc86b
3
+ size 6608313224
stats.json ADDED
@@ -0,0 +1,321 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "franka": {
3
+ "observation.state": {
4
+ "min": [
5
+ -0.4828203022480011,
6
+ -0.3255046010017395,
7
+ 0.008128180168569088,
8
+ 0.35277295112609863,
9
+ -3.641430377960205,
10
+ -1.842738389968872,
11
+ -0.0013586411951109767,
12
+ -0.042040832340717316
13
+ ],
14
+ "max": [
15
+ 0.21031762659549713,
16
+ 0.39128610491752625,
17
+ 1.3660105466842651,
18
+ 3.6714255809783936,
19
+ 3.560650587081909,
20
+ 1.386339545249939,
21
+ 0.04233968257904053,
22
+ 0.0013633022317662835
23
+ ],
24
+ "mean": [
25
+ -0.04651878401637077,
26
+ 0.034409064799547195,
27
+ 0.7645533084869385,
28
+ 2.9722108840942383,
29
+ -0.22046984732151031,
30
+ -0.12557929754257202,
31
+ 0.026914231479167938,
32
+ -0.02719080075621605
33
+ ],
34
+ "std": [
35
+ 0.10494393855333328,
36
+ 0.1517663300037384,
37
+ 0.37851467728614807,
38
+ 0.34427809715270996,
39
+ 0.906947135925293,
40
+ 0.3253919184207916,
41
+ 0.014175962656736374,
42
+ 0.014058873057365417
43
+ ],
44
+ "count": [
45
+ 273465
46
+ ]
47
+ },
48
+ "observation.images.wrist_image": {
49
+ "min": [
50
+ [
51
+ [
52
+ 0.0
53
+ ]
54
+ ],
55
+ [
56
+ [
57
+ 0.0
58
+ ]
59
+ ],
60
+ [
61
+ [
62
+ 0.0
63
+ ]
64
+ ]
65
+ ],
66
+ "max": [
67
+ [
68
+ [
69
+ 1.0
70
+ ]
71
+ ],
72
+ [
73
+ [
74
+ 1.0
75
+ ]
76
+ ],
77
+ [
78
+ [
79
+ 1.0
80
+ ]
81
+ ]
82
+ ],
83
+ "mean": [
84
+ [
85
+ [
86
+ 0.42383804087577415
87
+ ]
88
+ ],
89
+ [
90
+ [
91
+ 0.3673966255130251
92
+ ]
93
+ ],
94
+ [
95
+ [
96
+ 0.3188305260929261
97
+ ]
98
+ ]
99
+ ],
100
+ "std": [
101
+ [
102
+ [
103
+ 0.241981951960931
104
+ ]
105
+ ],
106
+ [
107
+ [
108
+ 0.2324922531543837
109
+ ]
110
+ ],
111
+ [
112
+ [
113
+ 0.22189874595427886
114
+ ]
115
+ ]
116
+ ],
117
+ "count": [
118
+ 37906
119
+ ]
120
+ },
121
+ "index": {
122
+ "min": [
123
+ 0.0
124
+ ],
125
+ "max": [
126
+ 101468.0
127
+ ],
128
+ "mean": [
129
+ 37110.32421875
130
+ ],
131
+ "std": [
132
+ 24809.974609375
133
+ ],
134
+ "count": [
135
+ 273465
136
+ ]
137
+ },
138
+ "action": {
139
+ "min": [
140
+ -0.9375,
141
+ -0.9375,
142
+ -0.9375,
143
+ -0.2582142949104309,
144
+ -0.375,
145
+ -0.3675000071525574,
146
+ 0.0
147
+ ],
148
+ "max": [
149
+ 0.9375,
150
+ 0.9375,
151
+ 0.9375,
152
+ 0.3557142913341522,
153
+ 0.375,
154
+ 0.375,
155
+ 1.0
156
+ ],
157
+ "mean": [
158
+ 0.06278158724308014,
159
+ 0.08684103190898895,
160
+ -0.09037306159734726,
161
+ 0.0005407430580817163,
162
+ 0.0056433891877532005,
163
+ -0.005229103844612837,
164
+ 0.5248202681541443
165
+ ],
166
+ "std": [
167
+ 0.33552372455596924,
168
+ 0.378447026014328,
169
+ 0.44472894072532654,
170
+ 0.03924354538321495,
171
+ 0.06339295208454132,
172
+ 0.07797033339738846,
173
+ 0.4993835687637329
174
+ ],
175
+ "count": [
176
+ 273465
177
+ ]
178
+ },
179
+ "observation.images.image": {
180
+ "min": [
181
+ [
182
+ [
183
+ 0.0
184
+ ]
185
+ ],
186
+ [
187
+ [
188
+ 0.0
189
+ ]
190
+ ],
191
+ [
192
+ [
193
+ 0.0
194
+ ]
195
+ ]
196
+ ],
197
+ "max": [
198
+ [
199
+ [
200
+ 1.0
201
+ ]
202
+ ],
203
+ [
204
+ [
205
+ 1.0
206
+ ]
207
+ ],
208
+ [
209
+ [
210
+ 1.0
211
+ ]
212
+ ]
213
+ ],
214
+ "mean": [
215
+ [
216
+ [
217
+ 0.40565344712473544
218
+ ]
219
+ ],
220
+ [
221
+ [
222
+ 0.3622734584518655
223
+ ]
224
+ ],
225
+ [
226
+ [
227
+ 0.3191169762835281
228
+ ]
229
+ ]
230
+ ],
231
+ "std": [
232
+ [
233
+ [
234
+ 0.24465104744671567
235
+ ]
236
+ ],
237
+ [
238
+ [
239
+ 0.24123842330511033
240
+ ]
241
+ ],
242
+ [
243
+ [
244
+ 0.23621162600478948
245
+ ]
246
+ ]
247
+ ],
248
+ "count": [
249
+ 37906
250
+ ]
251
+ },
252
+ "task_index": {
253
+ "min": [
254
+ 0.0
255
+ ],
256
+ "max": [
257
+ 9.0
258
+ ],
259
+ "mean": [
260
+ 4.499231338500977
261
+ ],
262
+ "std": [
263
+ 2.867191791534424
264
+ ],
265
+ "count": [
266
+ 273465
267
+ ]
268
+ },
269
+ "episode_index": {
270
+ "min": [
271
+ 0.0
272
+ ],
273
+ "max": [
274
+ 453.0
275
+ ],
276
+ "mean": [
277
+ 207.0758819580078
278
+ ],
279
+ "std": [
280
+ 121.92362976074219
281
+ ],
282
+ "count": [
283
+ 273465
284
+ ]
285
+ },
286
+ "frame_index": {
287
+ "min": [
288
+ 0.0
289
+ ],
290
+ "max": [
291
+ 504.0
292
+ ],
293
+ "mean": [
294
+ 94.6483154296875
295
+ ],
296
+ "std": [
297
+ 72.595703125
298
+ ],
299
+ "count": [
300
+ 273465
301
+ ]
302
+ },
303
+ "timestamp": {
304
+ "min": [
305
+ 0.0
306
+ ],
307
+ "max": [
308
+ 25.200000762939453
309
+ ],
310
+ "mean": [
311
+ 4.732414245605469
312
+ ],
313
+ "std": [
314
+ 3.6297781467437744
315
+ ],
316
+ "count": [
317
+ 273465
318
+ ]
319
+ }
320
+ }
321
+ }
train_config.json ADDED
@@ -0,0 +1,307 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "type": "cubev2",
4
+ "repo_id": "multidata_from_file",
5
+ "repo_id_file": "/home/jiangjiahao/data/ckpt/cubev2/_repo_id_files/cubev2-libero4-abs-chunk10-finetune-2026_04_17_13_42_47.txt",
6
+ "root": null,
7
+ "episodes": null,
8
+ "image_transforms": {
9
+ "enable": false,
10
+ "max_num_transforms": 3,
11
+ "random_order": false,
12
+ "tfs": {
13
+ "brightness": {
14
+ "weight": 1.0,
15
+ "type": "ColorJitter",
16
+ "kwargs": {
17
+ "brightness": [
18
+ 0.8,
19
+ 1.2
20
+ ]
21
+ }
22
+ },
23
+ "contrast": {
24
+ "weight": 1.0,
25
+ "type": "ColorJitter",
26
+ "kwargs": {
27
+ "contrast": [
28
+ 0.8,
29
+ 1.2
30
+ ]
31
+ }
32
+ },
33
+ "saturation": {
34
+ "weight": 1.0,
35
+ "type": "ColorJitter",
36
+ "kwargs": {
37
+ "saturation": [
38
+ 0.5,
39
+ 1.5
40
+ ]
41
+ }
42
+ },
43
+ "hue": {
44
+ "weight": 1.0,
45
+ "type": "ColorJitter",
46
+ "kwargs": {
47
+ "hue": [
48
+ -0.05,
49
+ 0.05
50
+ ]
51
+ }
52
+ },
53
+ "sharpness": {
54
+ "weight": 1.0,
55
+ "type": "SharpnessJitter",
56
+ "kwargs": {
57
+ "sharpness": [
58
+ 0.5,
59
+ 1.5
60
+ ]
61
+ }
62
+ },
63
+ "affine": {
64
+ "weight": 1.0,
65
+ "type": "RandomAffine",
66
+ "kwargs": {
67
+ "degrees": [
68
+ -5.0,
69
+ 5.0
70
+ ],
71
+ "translate": [
72
+ 0.05,
73
+ 0.05
74
+ ]
75
+ }
76
+ }
77
+ }
78
+ },
79
+ "revision": null,
80
+ "use_imagenet_stats": true,
81
+ "use_external_stats": true,
82
+ "external_stats_path": "/home/jiangjiahao/research/MagicBot-VGA/outputs/norm_stats/libero_all_chunk10/franka/abs/stats.json",
83
+ "external_stats_root": null,
84
+ "weight_rules_path": null,
85
+ "video_backend": "torchcodec",
86
+ "streaming": false,
87
+ "dist_loading": false,
88
+ "buffer_size": 1024,
89
+ "action_mode": "abs",
90
+ "repack_transforms": {
91
+ "inputs": [],
92
+ "outputs": []
93
+ },
94
+ "data_transforms": {
95
+ "inputs": [
96
+ {
97
+ "type": "resize_with_pad",
98
+ "height": 224,
99
+ "width": 224,
100
+ "mode": "bilinear"
101
+ },
102
+ {
103
+ "type": "remap_image_key",
104
+ "mapping": {}
105
+ },
106
+ {
107
+ "type": "normalize",
108
+ "selected_keys": null,
109
+ "mode": "mean_std",
110
+ "norm_stats": {}
111
+ },
112
+ {
113
+ "type": "compose_fields",
114
+ "mapping": {}
115
+ },
116
+ {
117
+ "type": "pad_state_and_action",
118
+ "max_state_dim": 32,
119
+ "max_action_dim": 32
120
+ },
121
+ {
122
+ "type": "cubev2_processor",
123
+ "pretrained_model_name_or_path": "/home/jiangjiahao/data/model/Qwen3-VL-2B-Instruct",
124
+ "max_length": 48,
125
+ "task_key": "task",
126
+ "padding_side": "right",
127
+ "padding": "max_length",
128
+ "truncation": true,
129
+ "spatial_merge_size": 2,
130
+ "vision_start_token_id": 151652,
131
+ "vision_end_token_id": 151653,
132
+ "image_token_id": 151655,
133
+ "processor": null,
134
+ "_processor_source": null
135
+ },
136
+ {
137
+ "type": "unify_cubev2_inputs"
138
+ }
139
+ ],
140
+ "outputs": []
141
+ },
142
+ "model_transforms": {
143
+ "inputs": [],
144
+ "outputs": []
145
+ },
146
+ "height": 224,
147
+ "width": 224,
148
+ "max_state_dim": 32,
149
+ "max_action_dim": 32,
150
+ "qwen3_vl_processor_path": "/home/jiangjiahao/data/model/Qwen3-VL-2B-Instruct"
151
+ },
152
+ "env": null,
153
+ "policy": {
154
+ "type": "cubev2",
155
+ "n_obs_steps": 1,
156
+ "input_features": {
157
+ "observation.state": {
158
+ "type": "STATE",
159
+ "shape": [
160
+ 32
161
+ ]
162
+ }
163
+ },
164
+ "output_features": {
165
+ "action": {
166
+ "type": "ACTION",
167
+ "shape": [
168
+ 32
169
+ ]
170
+ }
171
+ },
172
+ "device": "cuda:0",
173
+ "use_amp": false,
174
+ "push_to_hub": false,
175
+ "repo_id": "lerobot_lab/cubev2",
176
+ "private": null,
177
+ "tags": null,
178
+ "license": null,
179
+ "pretrained_path": "/home/jiangjiahao/data/model/MagicBot-VGA-Base",
180
+ "qwen3_vl_variant": "qwen3_vl_28l",
181
+ "action_expert_variant": "qwen3_28l",
182
+ "qwen3_vl_pretrained_path": "/home/jiangjiahao/data/model/Qwen3-VL-2B-Instruct",
183
+ "dtype": "bfloat16",
184
+ "chunk_size": 10,
185
+ "n_action_steps": 10,
186
+ "max_state_dim": 32,
187
+ "max_action_dim": 32,
188
+ "num_inference_steps": 10,
189
+ "time_sampling_beta_alpha": 1.5,
190
+ "time_sampling_beta_beta": 1.0,
191
+ "time_sampling_scale": 0.999,
192
+ "time_sampling_offset": 0.001,
193
+ "min_period": 0.004,
194
+ "max_period": 4.0,
195
+ "image_resolution": [
196
+ 224,
197
+ 224
198
+ ],
199
+ "empty_cameras": 0,
200
+ "normalization_mapping": {
201
+ "VISUAL": "IDENTITY",
202
+ "STATE": "IDENTITY",
203
+ "ACTION": "IDENTITY"
204
+ },
205
+ "gradient_checkpointing": false,
206
+ "compile_model": false,
207
+ "compile_mode": "max-autotune",
208
+ "optimizer_lr": 5e-05,
209
+ "optimizer_betas": [
210
+ 0.9,
211
+ 0.95
212
+ ],
213
+ "optimizer_eps": 1e-08,
214
+ "optimizer_weight_decay": 0.01,
215
+ "optimizer_grad_clip_norm": 1.0,
216
+ "scheduler_warmup_steps": 2000,
217
+ "scheduler_decay_steps": 80000,
218
+ "scheduler_decay_lr": 5e-06,
219
+ "tokenizer_max_length": 48,
220
+ "freeze_vision_encoder": false,
221
+ "train_expert_only": false,
222
+ "train_vlm_only": false,
223
+ "scale_factor": 8,
224
+ "lambda_gen": 0.01,
225
+ "cosmos_tokenizer_path_or_name": "/home/jiangjiahao/data/model/Cosmos-Tokenizer-CI8x8",
226
+ "enable_3d_queries": true,
227
+ "num_3d_query_tokens": 432,
228
+ "da3_alignment_mode": "query_decoder",
229
+ "da3_query_resampler_layers": 1,
230
+ "da3_query_resampler_ff_mult": 1,
231
+ "query_layer_indices": [
232
+ 13,
233
+ 19,
234
+ 23,
235
+ 27
236
+ ],
237
+ "da3_variant": "auto",
238
+ "da3_teacher_layers": [
239
+ 11,
240
+ 15,
241
+ 19,
242
+ 23
243
+ ],
244
+ "da3_query_dim": 2048,
245
+ "da3_tokens_per_view": 1296,
246
+ "da3_num_views": 3,
247
+ "lambda_3d": 0.01,
248
+ "da3_model_path_or_name": "/home/jiangjiahao/data/model/DA3-LARGE-1.1",
249
+ "da3_model_name": null,
250
+ "da3_code_root": null,
251
+ "da3_teacher_process_res": 504,
252
+ "da3_layer_weights": [
253
+ 1.0,
254
+ 1.2,
255
+ 1.4,
256
+ 1.6
257
+ ],
258
+ "future_query_init_std": 0.02,
259
+ "log_da3_teacher_timing": true
260
+ },
261
+ "output_dir": "/home/jiangjiahao/data/ckpt/cubev2/cubev2-libero4-abs-chunk10-finetune-2026_04_17_13_42_47",
262
+ "job_name": "cubev2-libero4-abs-chunk10-finetune-2026_04_17_13_42_47",
263
+ "resume": false,
264
+ "seed": 42,
265
+ "num_workers": 12,
266
+ "batch_size": 8,
267
+ "steps": 80000,
268
+ "eval_freq": 20000,
269
+ "log_freq": 25,
270
+ "save_checkpoint": true,
271
+ "save_freq": 10000,
272
+ "use_policy_training_preset": true,
273
+ "optimizer": {
274
+ "type": "adamw",
275
+ "lr": 5e-05,
276
+ "weight_decay": 0.01,
277
+ "grad_clip_norm": 1.0,
278
+ "betas": [
279
+ 0.9,
280
+ 0.95
281
+ ],
282
+ "eps": 1e-08
283
+ },
284
+ "scheduler": {
285
+ "type": "cosine_decay_with_warmup",
286
+ "num_warmup_steps": 2000,
287
+ "num_decay_steps": 80000,
288
+ "peak_lr": 5e-05,
289
+ "decay_lr": 5e-06
290
+ },
291
+ "eval": {
292
+ "n_episodes": 50,
293
+ "batch_size": 50,
294
+ "use_async_envs": false
295
+ },
296
+ "wandb": {
297
+ "enable": true,
298
+ "disable_artifact": false,
299
+ "project": "CUBEv2",
300
+ "entity": null,
301
+ "notes": null,
302
+ "run_id": "k7q0nz1k",
303
+ "mode": "online"
304
+ },
305
+ "checkpoint_path": null,
306
+ "rename_map": {}
307
+ }