HoloMotion
ONNX
Safetensors
kaihui-hug123 commited on
Commit
ad1765d
·
verified ·
1 Parent(s): 8df926b

Upload 2 files

Browse files
HoloMotion_motion_tracking_model_v1.3.2/config.yaml ADDED
@@ -0,0 +1,1726 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ project_name: HoloMotionMotrackV1.3
2
+ experiment_name: training/motion_tracking/release_internal_yaw_actor_obs_pos0p2_no_yaw_term_8x8GPUs
3
+ num_envs: 8192
4
+ headless: true
5
+ motion_h5_path: ???
6
+ checkpoint: null
7
+ num_processes: 64
8
+ main_process: true
9
+ process_id: 0
10
+ timestamp: ${now:%Y%m%d_%H%M%S}
11
+ base_dir: /horizon-bucket/robot_lab/users/bo07.zhang-labs/expr_output/HoloMotionMotionTracking
12
+ experiment_dir: ${base_dir}/${project_name}/${timestamp}-${experiment_name}
13
+ save_dir: ${experiment_dir}/.hydra
14
+ output_dir: ${experiment_dir}/output
15
+ experiment_save_dir: /horizon-bucket/robot_lab/users/bo07.zhang-labs/expr_output/HoloMotionMotionTracking/HoloMotionMotrackV1.3/20260602_141519-training/motion_tracking/release_internal_yaw_actor_obs_pos0p2_no_yaw_term_8x8GPUs
16
+ _group_:
17
+ enabled: false
18
+ model_type: holomotion
19
+ headless: false
20
+ record_video: false
21
+ video_width: 1280
22
+ video_height: 720
23
+ video_fps: 30
24
+ camera_tracking: true
25
+ camera_height_offset: 0.3
26
+ camera_distance: 4.0
27
+ camera_azimuth: 150.0
28
+ camera_elevation: -20.0
29
+ robot_xml_path: null
30
+ motion_npz_dir: null
31
+ motion_npz_path: null
32
+ ckpt_onnx_path: null
33
+ ckpt_onnx_root_dir: null
34
+ ckpt_onnx_names: null
35
+ dump_npzs: true
36
+ calc_per_clip_metrics: false
37
+ generate_report: false
38
+ metric_calculation: per_clip
39
+ dof_mode: '23'
40
+ failure_pos_err_thresh_m: 0.25
41
+ ray_actors_per_gpu: 16
42
+ ray_auto_reduce_actors_per_gpu: true
43
+ ray_actor_startup_probe_timeout_s: 300
44
+ ray_multi_ckpt_mode: split
45
+ use_gpu: true
46
+ algo:
47
+ _target_: holomotion.src.algo.ppo_tf.PPOTF
48
+ _recursive_: false
49
+ config:
50
+ enable_online_eval: false
51
+ num_learning_iterations: 10001
52
+ log_interval: 5
53
+ save_interval: 1000
54
+ export_policy: true
55
+ onnx_name_suffix: null
56
+ use_kv_cache: true
57
+ eval_interval: null
58
+ load_optimizer: true
59
+ headless: ${headless}
60
+ mixed_precision: null
61
+ dynamo_backend: inductor
62
+ init_at_random_ep_len: true
63
+ num_steps_per_env: 32
64
+ num_learning_epochs: 3
65
+ num_mini_batches: 24
66
+ clip_param: 0.2
67
+ gamma: 0.99
68
+ lam: 0.95
69
+ value_loss_coef: 1.0
70
+ entropy_coef: 0.005
71
+ anneal_entropy: false
72
+ zero_entropy_point: 1.0
73
+ max_grad_norm: 1.0
74
+ use_clipped_value_loss: true
75
+ desired_kl: 0.01
76
+ init_noise_std: 1.0
77
+ optimizer_type: AdamW
78
+ schedule: adaptive
79
+ actor_learning_rate: 3.0e-05
80
+ critic_learning_rate: 5.0e-05
81
+ adaptive_lr:
82
+ adapt_critic: false
83
+ lr_scaler: 1.2
84
+ kl_high_factor: 2.0
85
+ kl_low_factor: 0.5
86
+ min_learning_rate: 1.0e-06
87
+ max_learning_rate: 1.0
88
+ distributed_update:
89
+ mode: scalable
90
+ lr_scale:
91
+ mode: sqrt_world_size
92
+ reference_world_size: 1
93
+ max_scale: null
94
+ kl_early_stop:
95
+ enabled: true
96
+ signal: window_mean
97
+ window_size: 3
98
+ factor: 1.8
99
+ min_updates: 1
100
+ normalize_advantage_per_mini_batch: false
101
+ global_advantage_norm: true
102
+ sampling_strategy: weighted_bin
103
+ curriculum:
104
+ p_a_ratio: 0.5
105
+ ema_alpha_signal: 0.2
106
+ ema_alpha_rel_improve: 0.2
107
+ relative_eps: 1.0e-06
108
+ dump_whole_window_scores_json: false
109
+ dump_whole_window_scores_every_swaps: 10
110
+ weighted_bin:
111
+ bin_regex_patterns:
112
+ - regex: .*AMASS.*
113
+ ratio: 0.1
114
+ - regex: .*pico.*
115
+ ratio: 0.1
116
+ - regex: .*nyt.*
117
+ ratio: 0.1
118
+ - regex: .*sonic.*
119
+ ratio: 0.1
120
+ - regex: .*MotionMillion.*
121
+ ratio: 0.02
122
+ - regex: .*LAFAN1.*
123
+ ratio: 0.02
124
+ dump_sampled_keys: false
125
+ dump_sampled_keys_interval: 1000
126
+ module_dict:
127
+ actor: ${modules.actor}
128
+ critic: ${modules.critic}
129
+ symmetry_loss:
130
+ enabled: false
131
+ coef: 0.1
132
+ dof_sign_by_name: ${robot.dof_sign_by_name}
133
+ kl_coef: 0.0
134
+ noise_std_type: log
135
+ fix_sigma: false
136
+ min_sigma: 0.01
137
+ max_sigma: 1.2
138
+ aux_state_pred:
139
+ enabled: true
140
+ w_keybody_contact: 0.01
141
+ w_base_lin_vel: 0.01
142
+ w_ref_keybody_rel_pos: 0.1
143
+ w_robot_keybody_rel_pos: 0.1
144
+ min_std: 0.01
145
+ max_std: 2.0
146
+ keybody_contact_names:
147
+ - left_hip_pitch_link
148
+ - right_hip_pitch_link
149
+ - left_knee_link
150
+ - right_knee_link
151
+ - left_ankle_roll_link
152
+ - right_ankle_roll_link
153
+ - left_elbow_link
154
+ - right_elbow_link
155
+ - left_wrist_yaw_link
156
+ - right_wrist_yaw_link
157
+ keybody_rel_pos_names:
158
+ - left_knee_link
159
+ - right_knee_link
160
+ - left_ankle_roll_link
161
+ - right_ankle_roll_link
162
+ - left_elbow_link
163
+ - right_elbow_link
164
+ - left_wrist_yaw_link
165
+ - right_wrist_yaw_link
166
+ inactive_expert_margin_to_topk:
167
+ enabled: true
168
+ weight: 10
169
+ ratio_floor: 0.1
170
+ aux_router_command_recon:
171
+ enabled: false
172
+ weight: 0.0
173
+ hidden_dim: 0
174
+ term_prefix: actor_ref_
175
+ aux_router_switch_penalty:
176
+ enabled: false
177
+ weight: 0.0
178
+ router_expert_orthogonal:
179
+ enabled: false
180
+ weight: 0.0
181
+ min_active_usage: 0.001
182
+ eps: 1.0e-08
183
+ selected_expert_margin_to_unselected:
184
+ enabled: false
185
+ weight: 0.0
186
+ target: 0.0
187
+ moe_router:
188
+ routing_score_fn: softmax
189
+ routing_scale: 1.0
190
+ use_dynamic_bias: false
191
+ bias_update_rate: 0.001
192
+ expert_bias_clip: 0.0
193
+ robot:
194
+ humanoid_type: unitree/G1/29dof
195
+ dof_obs_size: 29
196
+ actions_dim: 29
197
+ num_bodies: 30
198
+ num_extend_bodies: 0
199
+ undesired_contacts_regrex: ^(?!left_ankle_roll_link$)(?!right_ankle_roll_link$)(?!left_wrist_yaw_link$)(?!right_wrist_yaw_link$).+$
200
+ torso_name: torso_link
201
+ anchor_body: torso_link
202
+ key_bodies:
203
+ - pelvis
204
+ - left_hip_roll_link
205
+ - left_knee_link
206
+ - left_ankle_pitch_link
207
+ - right_hip_roll_link
208
+ - right_knee_link
209
+ - right_ankle_pitch_link
210
+ - torso_link
211
+ - left_shoulder_roll_link
212
+ - left_elbow_link
213
+ - left_wrist_yaw_link
214
+ - right_shoulder_roll_link
215
+ - right_elbow_link
216
+ - right_wrist_yaw_link
217
+ key_dofs:
218
+ - left_knee_joint
219
+ - right_knee_joint
220
+ - left_elbow_joint
221
+ - right_elbow_joint
222
+ dof_names:
223
+ - left_hip_pitch_joint
224
+ - left_hip_roll_joint
225
+ - left_hip_yaw_joint
226
+ - left_knee_joint
227
+ - left_ankle_pitch_joint
228
+ - left_ankle_roll_joint
229
+ - right_hip_pitch_joint
230
+ - right_hip_roll_joint
231
+ - right_hip_yaw_joint
232
+ - right_knee_joint
233
+ - right_ankle_pitch_joint
234
+ - right_ankle_roll_joint
235
+ - waist_yaw_joint
236
+ - waist_roll_joint
237
+ - waist_pitch_joint
238
+ - left_shoulder_pitch_joint
239
+ - left_shoulder_roll_joint
240
+ - left_shoulder_yaw_joint
241
+ - left_elbow_joint
242
+ - left_wrist_roll_joint
243
+ - left_wrist_pitch_joint
244
+ - left_wrist_yaw_joint
245
+ - right_shoulder_pitch_joint
246
+ - right_shoulder_roll_joint
247
+ - right_shoulder_yaw_joint
248
+ - right_elbow_joint
249
+ - right_wrist_roll_joint
250
+ - right_wrist_pitch_joint
251
+ - right_wrist_yaw_joint
252
+ arm_dof_names:
253
+ - left_shoulder_pitch_joint
254
+ - left_shoulder_roll_joint
255
+ - left_shoulder_yaw_joint
256
+ - left_elbow_joint
257
+ - left_wrist_roll_joint
258
+ - left_wrist_pitch_joint
259
+ - left_wrist_yaw_joint
260
+ - right_shoulder_pitch_joint
261
+ - right_shoulder_roll_joint
262
+ - right_shoulder_yaw_joint
263
+ - right_elbow_joint
264
+ - right_wrist_roll_joint
265
+ - right_wrist_pitch_joint
266
+ - right_wrist_yaw_joint
267
+ waist_dof_names:
268
+ - waist_yaw_joint
269
+ - waist_roll_joint
270
+ - waist_pitch_joint
271
+ leg_dof_names:
272
+ - left_hip_pitch_joint
273
+ - left_hip_roll_joint
274
+ - left_hip_yaw_joint
275
+ - left_knee_joint
276
+ - left_ankle_pitch_joint
277
+ - left_ankle_roll_joint
278
+ - right_hip_pitch_joint
279
+ - right_hip_roll_joint
280
+ - right_hip_yaw_joint
281
+ - right_knee_joint
282
+ - right_ankle_pitch_joint
283
+ - right_ankle_roll_joint
284
+ left_arm_dof_names:
285
+ - left_shoulder_pitch_joint
286
+ - left_shoulder_roll_joint
287
+ - left_shoulder_yaw_joint
288
+ - left_elbow_joint
289
+ - left_wrist_roll_joint
290
+ - left_wrist_pitch_joint
291
+ - left_wrist_yaw_joint
292
+ right_arm_dof_names:
293
+ - right_shoulder_pitch_joint
294
+ - right_shoulder_roll_joint
295
+ - right_shoulder_yaw_joint
296
+ - right_elbow_joint
297
+ - right_wrist_roll_joint
298
+ - right_wrist_pitch_joint
299
+ - right_wrist_yaw_joint
300
+ left_leg_dof_names:
301
+ - left_hip_pitch_joint
302
+ - left_hip_roll_joint
303
+ - left_hip_yaw_joint
304
+ - left_knee_joint
305
+ - left_ankle_pitch_joint
306
+ - left_ankle_roll_joint
307
+ right_leg_dof_names:
308
+ - right_hip_pitch_joint
309
+ - right_hip_roll_joint
310
+ - right_hip_yaw_joint
311
+ - right_knee_joint
312
+ - right_ankle_pitch_joint
313
+ - right_ankle_roll_joint
314
+ upper_body_dof_names: ${robot.arm_dof_names}
315
+ lower_body_dof_names:
316
+ - left_hip_pitch_joint
317
+ - left_hip_roll_joint
318
+ - left_hip_yaw_joint
319
+ - left_knee_joint
320
+ - left_ankle_pitch_joint
321
+ - left_ankle_roll_joint
322
+ - right_hip_pitch_joint
323
+ - right_hip_roll_joint
324
+ - right_hip_yaw_joint
325
+ - right_knee_joint
326
+ - right_ankle_pitch_joint
327
+ - right_ankle_roll_joint
328
+ - waist_yaw_joint
329
+ - waist_roll_joint
330
+ - waist_pitch_joint
331
+ arm_body_names:
332
+ - left_shoulder_pitch_link
333
+ - left_shoulder_roll_link
334
+ - left_shoulder_yaw_link
335
+ - left_elbow_link
336
+ - left_wrist_roll_link
337
+ - left_wrist_pitch_link
338
+ - left_wrist_yaw_link
339
+ - right_shoulder_pitch_link
340
+ - right_shoulder_roll_link
341
+ - right_shoulder_yaw_link
342
+ - right_elbow_link
343
+ - right_wrist_roll_link
344
+ - right_wrist_pitch_link
345
+ - right_wrist_yaw_link
346
+ head_hand_bodies:
347
+ - torso_link
348
+ - left_wrist_yaw_link
349
+ - right_wrist_yaw_link
350
+ torso_body_names:
351
+ - waist_yaw_link
352
+ - waist_roll_link
353
+ - torso_link
354
+ leg_body_names:
355
+ - left_hip_pitch_link
356
+ - left_hip_roll_link
357
+ - left_hip_yaw_link
358
+ - left_knee_link
359
+ - left_ankle_pitch_link
360
+ - left_ankle_roll_link
361
+ - right_hip_pitch_link
362
+ - right_hip_roll_link
363
+ - right_hip_yaw_link
364
+ - right_knee_link
365
+ - right_ankle_pitch_link
366
+ - right_ankle_roll_link
367
+ left_arm_body_names:
368
+ - left_shoulder_pitch_link
369
+ - left_shoulder_roll_link
370
+ - left_shoulder_yaw_link
371
+ - left_elbow_link
372
+ - left_wrist_roll_link
373
+ - left_wrist_pitch_link
374
+ - left_wrist_yaw_link
375
+ right_arm_body_names:
376
+ - right_shoulder_pitch_link
377
+ - right_shoulder_roll_link
378
+ - right_shoulder_yaw_link
379
+ - right_elbow_link
380
+ - right_wrist_roll_link
381
+ - right_wrist_pitch_link
382
+ - right_wrist_yaw_link
383
+ left_leg_body_names:
384
+ - left_hip_pitch_link
385
+ - left_hip_roll_link
386
+ - left_hip_yaw_link
387
+ - left_knee_link
388
+ - left_ankle_pitch_link
389
+ - left_ankle_roll_link
390
+ right_leg_body_names:
391
+ - right_hip_pitch_link
392
+ - right_hip_roll_link
393
+ - right_hip_yaw_link
394
+ - right_knee_link
395
+ - right_ankle_pitch_link
396
+ - right_ankle_roll_link
397
+ body_names:
398
+ - pelvis
399
+ - left_hip_pitch_link
400
+ - left_hip_roll_link
401
+ - left_hip_yaw_link
402
+ - left_knee_link
403
+ - left_ankle_pitch_link
404
+ - left_ankle_roll_link
405
+ - right_hip_pitch_link
406
+ - right_hip_roll_link
407
+ - right_hip_yaw_link
408
+ - right_knee_link
409
+ - right_ankle_pitch_link
410
+ - right_ankle_roll_link
411
+ - waist_yaw_link
412
+ - waist_roll_link
413
+ - torso_link
414
+ - left_shoulder_pitch_link
415
+ - left_shoulder_roll_link
416
+ - left_shoulder_yaw_link
417
+ - left_elbow_link
418
+ - left_wrist_roll_link
419
+ - left_wrist_pitch_link
420
+ - left_wrist_yaw_link
421
+ - right_shoulder_pitch_link
422
+ - right_shoulder_roll_link
423
+ - right_shoulder_yaw_link
424
+ - right_elbow_link
425
+ - right_wrist_roll_link
426
+ - right_wrist_pitch_link
427
+ - right_wrist_yaw_link
428
+ init_state:
429
+ pos:
430
+ - 0.0
431
+ - 0.0
432
+ - 0.8
433
+ rot:
434
+ - 0.0
435
+ - 0.929
436
+ - 0.341
437
+ - 0.298
438
+ lin_vel:
439
+ - 0.0
440
+ - 0.0
441
+ - 0.0
442
+ ang_vel:
443
+ - 0.0
444
+ - 0.0
445
+ - 0.0
446
+ default_joint_angles:
447
+ left_hip_pitch_joint: -0.312
448
+ left_hip_roll_joint: 0.0
449
+ left_hip_yaw_joint: 0.0
450
+ left_knee_joint: 0.669
451
+ left_ankle_pitch_joint: -0.363
452
+ left_ankle_roll_joint: 0.0
453
+ right_hip_pitch_joint: -0.312
454
+ right_hip_roll_joint: 0.0
455
+ right_hip_yaw_joint: 0.0
456
+ right_knee_joint: 0.669
457
+ right_ankle_pitch_joint: -0.363
458
+ right_ankle_roll_joint: 0.0
459
+ waist_yaw_joint: 0.0
460
+ waist_roll_joint: 0.0
461
+ waist_pitch_joint: 0.1
462
+ left_shoulder_pitch_joint: 0.2
463
+ left_shoulder_roll_joint: 0.2
464
+ left_shoulder_yaw_joint: 0.0
465
+ left_elbow_joint: 0.6
466
+ left_wrist_roll_joint: 0.0
467
+ left_wrist_pitch_joint: 0.0
468
+ left_wrist_yaw_joint: 0.0
469
+ right_shoulder_pitch_joint: 0.2
470
+ right_shoulder_roll_joint: -0.2
471
+ right_shoulder_yaw_joint: 0.0
472
+ right_elbow_joint: 0.6
473
+ right_wrist_roll_joint: 0.0
474
+ right_wrist_pitch_joint: 0.0
475
+ right_wrist_yaw_joint: 0.0
476
+ actuators:
477
+ actuator_type: unitree_erfi
478
+ ema_filter_enabled: false
479
+ ema_filter_alpha: 1.0
480
+ all_joints:
481
+ joint_names_expr:
482
+ - .*_hip_yaw_joint
483
+ - .*_hip_roll_joint
484
+ - .*_hip_pitch_joint
485
+ - .*_knee_joint
486
+ - .*_ankle_pitch_joint
487
+ - .*_ankle_roll_joint
488
+ - waist_yaw_joint
489
+ - waist_roll_joint
490
+ - waist_pitch_joint
491
+ - .*_shoulder_pitch_joint
492
+ - .*_shoulder_roll_joint
493
+ - .*_shoulder_yaw_joint
494
+ - .*_elbow_joint
495
+ - .*_wrist_roll_joint
496
+ - .*_wrist_pitch_joint
497
+ - .*_wrist_yaw_joint
498
+ effort_limit_sim:
499
+ .*_hip_yaw_joint: 88.0
500
+ .*_hip_roll_joint: 139.0
501
+ .*_hip_pitch_joint: 88.0
502
+ .*_knee_joint: 139.0
503
+ .*_ankle_pitch_joint: 50.0
504
+ .*_ankle_roll_joint: 50.0
505
+ waist_yaw_joint: 88.0
506
+ waist_roll_joint: 50.0
507
+ waist_pitch_joint: 50.0
508
+ .*_shoulder_pitch_joint: 25.0
509
+ .*_shoulder_roll_joint: 25.0
510
+ .*_shoulder_yaw_joint: 25.0
511
+ .*_elbow_joint: 25.0
512
+ .*_wrist_roll_joint: 25.0
513
+ .*_wrist_pitch_joint: 5.0
514
+ .*_wrist_yaw_joint: 5.0
515
+ velocity_limit_sim:
516
+ .*_hip_yaw_joint: 32.0
517
+ .*_hip_roll_joint: 20.0
518
+ .*_hip_pitch_joint: 32.0
519
+ .*_knee_joint: 20.0
520
+ .*_ankle_pitch_joint: 37.0
521
+ .*_ankle_roll_joint: 37.0
522
+ waist_yaw_joint: 32.0
523
+ waist_roll_joint: 37.0
524
+ waist_pitch_joint: 37.0
525
+ .*_shoulder_pitch_joint: 37.0
526
+ .*_shoulder_roll_joint: 37.0
527
+ .*_shoulder_yaw_joint: 37.0
528
+ .*_elbow_joint: 37.0
529
+ .*_wrist_roll_joint: 37.0
530
+ .*_wrist_pitch_joint: 22.0
531
+ .*_wrist_yaw_joint: 22.0
532
+ stiffness:
533
+ .*_hip_pitch_joint: 40.17923847
534
+ .*_hip_roll_joint: 99.09842778
535
+ .*_hip_yaw_joint: 40.17923847
536
+ .*_knee_joint: 99.09842778
537
+ .*_ankle_pitch_joint: 28.5012462
538
+ .*_ankle_roll_joint: 28.5012462
539
+ waist_yaw_joint: 40.17923847
540
+ waist_roll_joint: 28.5012462
541
+ waist_pitch_joint: 28.5012462
542
+ .*_shoulder_pitch_joint: 14.2506231
543
+ .*_shoulder_roll_joint: 14.2506231
544
+ .*_shoulder_yaw_joint: 14.2506231
545
+ .*_elbow_joint: 14.2506231
546
+ .*_wrist_roll_joint: 14.25062309787429
547
+ .*_wrist_pitch_joint: 16.77832748089279
548
+ .*_wrist_yaw_joint: 16.77832748089279
549
+ damping:
550
+ .*_hip_pitch_joint: 2.55788977
551
+ .*_hip_roll_joint: 6.30880185
552
+ .*_hip_yaw_joint: 2.55788977
553
+ .*_knee_joint: 6.30880185
554
+ .*_ankle_pitch_joint: 1.81444569
555
+ .*_ankle_roll_joint: 1.81444569
556
+ waist_yaw_joint: 2.55788977
557
+ waist_roll_joint: 1.81444569
558
+ waist_pitch_joint: 1.81444569
559
+ .*_shoulder_pitch_joint: 0.90722284
560
+ .*_shoulder_roll_joint: 0.90722284
561
+ .*_shoulder_yaw_joint: 0.90722284
562
+ .*_elbow_joint: 0.90722284
563
+ .*_wrist_roll_joint: 0.907222843292423
564
+ .*_wrist_pitch_joint: 1.06814150219
565
+ .*_wrist_yaw_joint: 1.06814150219
566
+ armature:
567
+ .*_hip_pitch_joint: 0.01017752
568
+ .*_hip_roll_joint: 0.025101925
569
+ .*_hip_yaw_joint: 0.01017752
570
+ .*_knee_joint: 0.025101925
571
+ .*_ankle_pitch_joint: 0.00721945
572
+ .*_ankle_roll_joint: 0.00721945
573
+ waist_yaw_joint: 0.01017752
574
+ waist_roll_joint: 0.00721945
575
+ waist_pitch_joint: 0.00721945
576
+ .*_shoulder_pitch_joint: 0.003609725
577
+ .*_shoulder_roll_joint: 0.003609725
578
+ .*_shoulder_yaw_joint: 0.003609725
579
+ .*_elbow_joint: 0.003609725
580
+ .*_wrist_roll_joint: 0.003609725
581
+ .*_wrist_pitch_joint: 0.00425
582
+ .*_wrist_yaw_joint: 0.00425
583
+ action_scale:
584
+ .*_hip_pitch_joint: 0.548
585
+ .*_hip_roll_joint: 0.351
586
+ .*_hip_yaw_joint: 0.548
587
+ .*_knee_joint: 0.351
588
+ .*_ankle_pitch_joint: 0.439
589
+ .*_ankle_roll_joint: 0.439
590
+ waist_yaw_joint: 0.548
591
+ waist_roll_joint: 0.439
592
+ waist_pitch_joint: 0.439
593
+ .*_shoulder_pitch_joint: 0.439
594
+ .*_shoulder_roll_joint: 0.439
595
+ .*_shoulder_yaw_joint: 0.439
596
+ .*_elbow_joint: 0.439
597
+ .*_wrist_roll_joint: 0.439
598
+ .*_wrist_pitch_joint: 0.075
599
+ .*_wrist_yaw_joint: 0.075
600
+ dof_sign_by_name:
601
+ left_hip_pitch_joint: 1.0
602
+ left_hip_roll_joint: -1.0
603
+ left_hip_yaw_joint: -1.0
604
+ left_knee_joint: 1.0
605
+ left_ankle_pitch_joint: 1.0
606
+ left_ankle_roll_joint: -1.0
607
+ right_hip_pitch_joint: 1.0
608
+ right_hip_roll_joint: -1.0
609
+ right_hip_yaw_joint: -1.0
610
+ right_knee_joint: 1.0
611
+ right_ankle_pitch_joint: 1.0
612
+ right_ankle_roll_joint: -1.0
613
+ waist_yaw_joint: -1.0
614
+ waist_roll_joint: -1.0
615
+ waist_pitch_joint: 1.0
616
+ left_shoulder_pitch_joint: 1.0
617
+ left_shoulder_roll_joint: -1.0
618
+ left_shoulder_yaw_joint: -1.0
619
+ left_elbow_joint: 1.0
620
+ left_wrist_roll_joint: -1.0
621
+ left_wrist_pitch_joint: 1.0
622
+ left_wrist_yaw_joint: -1.0
623
+ right_shoulder_pitch_joint: 1.0
624
+ right_shoulder_roll_joint: -1.0
625
+ right_shoulder_yaw_joint: -1.0
626
+ right_elbow_joint: 1.0
627
+ right_wrist_roll_joint: -1.0
628
+ right_wrist_pitch_joint: 1.0
629
+ right_wrist_yaw_joint: -1.0
630
+ asset:
631
+ collapse_fixed_joints: true
632
+ replace_cylinder_with_capsule: true
633
+ flip_visual_attachments: false
634
+ max_angular_velocity: 1000.0
635
+ max_linear_velocity: 1000.0
636
+ density: 0.001
637
+ angular_damping: 0.0
638
+ linear_damping: 0.0
639
+ asset_root: ./
640
+ urdf_file: assets/robots/${robot.humanoid_type}/g1_29dof_rev_1_0.urdf
641
+ assetFileName: assets/robots/${robot.humanoid_type}/g1_29dof_rev_1_0.xml
642
+ fix_base_link: false
643
+ force_usd_conversion: true
644
+ extend_config: []
645
+ motion:
646
+ asset:
647
+ assetRoot: ./
648
+ assetFileName: assets/robots/${robot.humanoid_type}/g1_29dof_rev_1_0.xml
649
+ sampling_strategy: ${algo.config.sampling_strategy}
650
+ weighted_bin: ${algo.config.weighted_bin}
651
+ curriculum: ${algo.config.curriculum}
652
+ dump_sampled_motion_keys: false
653
+ dump_sampled_motion_keys_interval: 1
654
+ dump_sampled_motion_keys_dir: sampled_motion_cache_keys
655
+ max_frame_length: 300
656
+ min_frame_length: 50
657
+ handpicked_motion_names: null
658
+ excluded_motion_names: null
659
+ world_frame_normalization: true
660
+ backend: hdf5_v2
661
+ train_hdf5_roots: ${train_hdf5_roots}
662
+ val_hdf5_roots: ${train_hdf5_roots}
663
+ dof_names: ${robot.dof_names}
664
+ body_names: ${robot.body_names}
665
+ key_bodies: ${robot.key_bodies}
666
+ extend_config: ${robot.extend_config}
667
+ dataloader:
668
+ num_workers: 2
669
+ prefetch_factor: 1
670
+ pin_memory: true
671
+ persistent_workers: false
672
+ timeout: 600
673
+ fk_robot_file_path: ${robot.asset.urdf_file}
674
+ fk_vel_smoothing_sigma: 2.0
675
+ online_filter:
676
+ enabled: false
677
+ butter_order: 4
678
+ butter_cutoff_hz_pool: []
679
+ cache:
680
+ batch_progress_bar: false
681
+ max_num_clips: ${num_envs}
682
+ device: cuda
683
+ swap_interval_steps: ${robot.motion.max_frame_length}
684
+ allowed_prefixes:
685
+ - ref_
686
+ - ft_ref_
687
+ env:
688
+ _target_: holomotion.src.env.motion_tracking.MotionTrackingEnv
689
+ _recursive_: false
690
+ config:
691
+ experiment_name: ${experiment_name}
692
+ num_envs: ${num_envs}
693
+ env_spacing: 2.5
694
+ replicate_physics: true
695
+ headless: ${headless}
696
+ num_processes: ${num_processes}
697
+ main_process: ${main_process}
698
+ process_id: ${process_id}
699
+ ckpt_dir: null
700
+ disable_ref_viz: false
701
+ eval_log_dir: null
702
+ save_rendering_dir: /horizon-bucket/robot_lab/users/bo07.zhang-labs/expr_output/HoloMotionMotionTracking/HoloMotionMotrackV1.3/20260602_141519-training/motion_tracking/release_internal_yaw_actor_obs_pos0p2_no_yaw_term_8x8GPUs/renderings_training
703
+ robot: ${robot}
704
+ domain_rand: ${domain_rand}
705
+ rewards: ${rewards}
706
+ terrain: ${terrain}
707
+ obs: ${obs}
708
+ terminations: ${terminations}
709
+ simulation:
710
+ episode_length_s: 10
711
+ sim_freq: 200
712
+ control_decimation: 4
713
+ physx:
714
+ bounce_threshold_velocity: 0.5
715
+ gpu_max_rigid_patch_count: 327680
716
+ scene:
717
+ terrain: ${terrain}
718
+ lighting:
719
+ distant_light_intensity: 3000.0
720
+ dome_light_intensity: 1000.0
721
+ contact_sensor:
722
+ history_length: 3
723
+ force_threshold: 10.0
724
+ track_air_time: true
725
+ debug_vis: false
726
+ actions:
727
+ dof_pos:
728
+ type: joint_position
729
+ params:
730
+ asset_name: robot
731
+ joint_names:
732
+ - .*
733
+ use_default_offset: true
734
+ scale: ${robot.actuators.action_scale}
735
+ commands:
736
+ ref_motion:
737
+ type: MotionCommandCfg
738
+ params:
739
+ command_obs_name: bydmmc_ref_motion
740
+ motion_lib_cfg: ${robot.motion}
741
+ urdf_dof_names: ${robot.dof_names}
742
+ urdf_body_names: ${robot.body_names}
743
+ arm_dof_names: ${robot.arm_dof_names}
744
+ waist_dof_names: ${robot.waist_dof_names}
745
+ leg_dof_names: ${robot.leg_dof_names}
746
+ arm_body_names: ${robot.arm_body_names}
747
+ torso_body_names: ${robot.torso_body_names}
748
+ leg_body_names: ${robot.leg_body_names}
749
+ anchor_bodylink_name: ${robot.anchor_body}
750
+ asset_name: robot
751
+ debug_vis: true
752
+ root_pose_perturb_range: ${domain_rand.motion_init_perturb.root_pose_perturb_range}
753
+ root_vel_perturb_range: ${domain_rand.motion_init_perturb.root_vel_perturb_range}
754
+ dof_pos_perturb_range: ${domain_rand.motion_init_perturb.dof_pos_perturb_range}
755
+ dof_vel_perturb_range: ${domain_rand.motion_init_perturb.dof_vel_perturb_range}
756
+ resample_time_interval_s: 100
757
+ n_fut_frames: ${obs.n_fut_frames}
758
+ target_fps: 50
759
+ normalization:
760
+ clip_actions: 100.0
761
+ clip_observations: 100.0
762
+ resample_motion_when_training: true
763
+ curriculum:
764
+ enabled: false
765
+ robot_friction_completion_rate:
766
+ enabled: true
767
+ func: robot_friction_range_by_completion_rate
768
+ params:
769
+ num_updates: 5
770
+ cr_thresholds:
771
+ - 0.1
772
+ - 0.2
773
+ - 0.28
774
+ - 0.34
775
+ - 0.4
776
+ static_friction_target:
777
+ - 0.3
778
+ - 1.6
779
+ dynamic_friction_target:
780
+ - 0.3
781
+ - 1.2
782
+ body_names: .*
783
+ restitution_range:
784
+ - 0.0
785
+ - 0.5
786
+ num_buckets: 64
787
+ rigid_body_com_completion_rate:
788
+ enabled: true
789
+ func: rigid_body_com_by_completion_rate
790
+ params:
791
+ num_updates: 5
792
+ cr_thresholds:
793
+ - 0.1
794
+ - 0.2
795
+ - 0.28
796
+ - 0.34
797
+ - 0.4
798
+ state_prefix: _cr_curr
799
+ asset_name: robot
800
+ body_names: torso_link
801
+ com_range_target:
802
+ x:
803
+ - -0.025
804
+ - 0.025
805
+ 'y':
806
+ - -0.05
807
+ - 0.05
808
+ z:
809
+ - -0.05
810
+ - 0.05
811
+ default_dof_pos_bias_completion_rate:
812
+ enabled: true
813
+ func: default_dof_pos_bias_by_completion_rate
814
+ params:
815
+ num_updates: 5
816
+ cr_thresholds:
817
+ - 0.1
818
+ - 0.2
819
+ - 0.28
820
+ - 0.34
821
+ - 0.4
822
+ state_prefix: _cr_curr
823
+ joint_names:
824
+ - .*
825
+ pos_distribution_params_target:
826
+ - -0.01
827
+ - 0.01
828
+ operation: add
829
+ distribution: uniform
830
+ push_by_setting_velocity_completion_rate:
831
+ enabled: true
832
+ func: isaaclab_mdp.modify_term_cfg
833
+ params:
834
+ address: events.push_by_setting_velocity.params
835
+ modify_fn: push_by_setting_velocity_range_by_completion_rate
836
+ modify_params:
837
+ num_updates: 3
838
+ cr_thresholds:
839
+ - 0.2
840
+ - 0.3
841
+ - 0.4
842
+ velocity_range_target:
843
+ x:
844
+ - -0.5
845
+ - 0.5
846
+ 'y':
847
+ - -0.5
848
+ - 0.5
849
+ z:
850
+ - -0.2
851
+ - 0.2
852
+ roll:
853
+ - -0.52
854
+ - 0.52
855
+ pitch:
856
+ - -0.52
857
+ - 0.52
858
+ yaw:
859
+ - -0.78
860
+ - 0.78
861
+ randomize_actuator_gains_completion_rate:
862
+ enabled: true
863
+ func: randomize_actuator_gains_by_completion_rate
864
+ params:
865
+ num_updates: 3
866
+ cr_thresholds:
867
+ - 0.2
868
+ - 0.3
869
+ - 0.4
870
+ asset_name: robot
871
+ body_names: .*
872
+ stiffness_distribution_params_target:
873
+ - 0.9
874
+ - 1.1
875
+ damping_distribution_params_target:
876
+ - 0.9
877
+ - 1.1
878
+ operation: scale
879
+ distribution: uniform
880
+ action_rate_l2_completion_rate:
881
+ enabled: true
882
+ func: reward_term_weight_by_completion_rate
883
+ params:
884
+ reward_term_name: action_rate_l2
885
+ final_weight: -0.1
886
+ start_scale: 0.1
887
+ num_updates: 5
888
+ cr_thresholds:
889
+ - 0.1
890
+ - 0.2
891
+ - 0.28
892
+ - 0.34
893
+ - 0.4
894
+ joint_pos_limits_completion_rate:
895
+ enabled: true
896
+ func: reward_term_weight_by_completion_rate
897
+ params:
898
+ reward_term_name: joint_pos_limits
899
+ final_weight: -10.0
900
+ start_scale: 0.1
901
+ num_updates: 5
902
+ cr_thresholds:
903
+ - 0.1
904
+ - 0.2
905
+ - 0.28
906
+ - 0.34
907
+ - 0.4
908
+ undesired_contacts_completion_rate:
909
+ enabled: true
910
+ func: reward_term_weight_by_completion_rate
911
+ params:
912
+ reward_term_name: undesired_contacts
913
+ final_weight: -0.1
914
+ start_scale: 0.1
915
+ num_updates: 5
916
+ cr_thresholds:
917
+ - 0.1
918
+ - 0.2
919
+ - 0.28
920
+ - 0.34
921
+ - 0.4
922
+ terminations:
923
+ time_out:
924
+ time_out: true
925
+ ref_gravity_projection_far:
926
+ params:
927
+ threshold: 0.8
928
+ ref_prefix: ${rewards._config.reward_prefix}
929
+ keybody_ref_z_far:
930
+ params:
931
+ threshold: 0.25
932
+ ref_prefix: ${rewards._config.reward_prefix}
933
+ keybody_names:
934
+ - pelvis
935
+ - left_ankle_roll_link
936
+ - right_ankle_roll_link
937
+ - left_wrist_yaw_link
938
+ - right_wrist_yaw_link
939
+ keybody_ref_pos_far:
940
+ params:
941
+ threshold: 0.2
942
+ ref_prefix: ${rewards._config.reward_prefix}
943
+ keybody_names:
944
+ - pelvis
945
+ obs:
946
+ context_length: 1
947
+ n_fut_frames: 10
948
+ target_fps: 50
949
+ actor_obs_prefix: ref_
950
+ critic_obs_prefix: ref_
951
+ obs_groups:
952
+ unified:
953
+ atomic_obs_list:
954
+ - actor_ref_gravity_projection_cur:
955
+ func: ref_gravity_projection_cur
956
+ history_length: ${obs.context_length}
957
+ flatten_history_dim: false
958
+ params:
959
+ ref_prefix: ${obs.actor_obs_prefix}
960
+ noise:
961
+ type: AdditiveUniformNoiseCfg
962
+ params:
963
+ n_min: ${domain_rand.obs_noise.actor_ref_gravity_projection_cur.n_min}
964
+ n_max: ${domain_rand.obs_noise.actor_ref_gravity_projection_cur.n_max}
965
+ - actor_ref_gravity_projection_fut:
966
+ func: ref_gravity_projection_fut
967
+ params:
968
+ ref_prefix: ${obs.actor_obs_prefix}
969
+ num_frames: ${obs.n_fut_frames}
970
+ noise:
971
+ type: AdditiveUniformNoiseCfg
972
+ params:
973
+ n_min: ${domain_rand.obs_noise.actor_ref_gravity_projection_fut.n_min}
974
+ n_max: ${domain_rand.obs_noise.actor_ref_gravity_projection_fut.n_max}
975
+ - actor_ref_base_linvel_cur:
976
+ func: ref_base_linvel_cur
977
+ history_length: ${obs.context_length}
978
+ flatten_history_dim: false
979
+ params:
980
+ ref_prefix: ${obs.actor_obs_prefix}
981
+ noise:
982
+ type: AdditiveUniformNoiseCfg
983
+ params:
984
+ n_min: ${domain_rand.obs_noise.actor_ref_base_linvel_cur.n_min}
985
+ n_max: ${domain_rand.obs_noise.actor_ref_base_linvel_cur.n_max}
986
+ n_min_z: ${domain_rand.obs_noise.actor_ref_base_linvel_cur.n_min_z}
987
+ n_max_z: ${domain_rand.obs_noise.actor_ref_base_linvel_cur.n_max_z}
988
+ - actor_ref_base_linvel_fut:
989
+ func: ref_base_linvel_fut
990
+ params:
991
+ ref_prefix: ${obs.actor_obs_prefix}
992
+ num_frames: ${obs.n_fut_frames}
993
+ noise:
994
+ type: AdditiveUniformNoiseCfg
995
+ params:
996
+ n_min: ${domain_rand.obs_noise.actor_ref_base_linvel_fut.n_min}
997
+ n_max: ${domain_rand.obs_noise.actor_ref_base_linvel_fut.n_max}
998
+ n_min_z: ${domain_rand.obs_noise.actor_ref_base_linvel_fut.n_min_z}
999
+ n_max_z: ${domain_rand.obs_noise.actor_ref_base_linvel_fut.n_max_z}
1000
+ - actor_ref_base_angvel_cur:
1001
+ func: ref_base_angvel_cur
1002
+ history_length: ${obs.context_length}
1003
+ flatten_history_dim: false
1004
+ params:
1005
+ ref_prefix: ${obs.actor_obs_prefix}
1006
+ noise:
1007
+ type: AdditiveUniformNoiseCfg
1008
+ params:
1009
+ n_min: ${domain_rand.obs_noise.actor_ref_base_angvel_cur.n_min}
1010
+ n_max: ${domain_rand.obs_noise.actor_ref_base_angvel_cur.n_max}
1011
+ n_min_z: ${domain_rand.obs_noise.actor_ref_base_angvel_cur.n_min_z}
1012
+ n_max_z: ${domain_rand.obs_noise.actor_ref_base_angvel_cur.n_max_z}
1013
+ - actor_ref_base_angvel_fut:
1014
+ func: ref_base_angvel_fut
1015
+ params:
1016
+ ref_prefix: ${obs.actor_obs_prefix}
1017
+ num_frames: ${obs.n_fut_frames}
1018
+ noise:
1019
+ type: AdditiveUniformNoiseCfg
1020
+ params:
1021
+ n_min: ${domain_rand.obs_noise.actor_ref_base_angvel_fut.n_min}
1022
+ n_max: ${domain_rand.obs_noise.actor_ref_base_angvel_fut.n_max}
1023
+ n_min_z: ${domain_rand.obs_noise.actor_ref_base_angvel_fut.n_min_z}
1024
+ n_max_z: ${domain_rand.obs_noise.actor_ref_base_angvel_fut.n_max_z}
1025
+ - actor_ref_dof_pos_cur:
1026
+ func: ref_dof_pos_cur
1027
+ history_length: ${obs.context_length}
1028
+ flatten_history_dim: false
1029
+ params:
1030
+ ref_prefix: ${obs.actor_obs_prefix}
1031
+ noise:
1032
+ type: AdditiveUniformNoiseCfg
1033
+ params:
1034
+ n_min: ${domain_rand.obs_noise.actor_ref_dof_pos_cur.n_min}
1035
+ n_max: ${domain_rand.obs_noise.actor_ref_dof_pos_cur.n_max}
1036
+ - actor_ref_dof_pos_fut:
1037
+ func: ref_dof_pos_fut
1038
+ params:
1039
+ ref_prefix: ${obs.actor_obs_prefix}
1040
+ num_frames: ${obs.n_fut_frames}
1041
+ noise:
1042
+ type: AdditiveUniformNoiseCfg
1043
+ params:
1044
+ n_min: ${domain_rand.obs_noise.actor_ref_dof_pos_fut.n_min}
1045
+ n_max: ${domain_rand.obs_noise.actor_ref_dof_pos_fut.n_max}
1046
+ - actor_ref_motion_filter_cutoff_hz:
1047
+ func: ref_motion_filter_cutoff_hz
1048
+ - actor_ref_root_height_cur:
1049
+ func: ref_root_height_cur
1050
+ history_length: ${obs.context_length}
1051
+ flatten_history_dim: false
1052
+ params:
1053
+ ref_prefix: ${obs.actor_obs_prefix}
1054
+ noise:
1055
+ type: AdditiveUniformNoiseCfg
1056
+ params:
1057
+ n_min: ${domain_rand.obs_noise.actor_ref_root_height_cur.n_min}
1058
+ n_max: ${domain_rand.obs_noise.actor_ref_root_height_cur.n_max}
1059
+ - actor_ref_root_height_fut:
1060
+ func: ref_root_height_fut
1061
+ params:
1062
+ ref_prefix: ${obs.actor_obs_prefix}
1063
+ num_frames: ${obs.n_fut_frames}
1064
+ noise:
1065
+ type: AdditiveUniformNoiseCfg
1066
+ params:
1067
+ n_min: ${domain_rand.obs_noise.actor_ref_root_height_fut.n_min}
1068
+ n_max: ${domain_rand.obs_noise.actor_ref_root_height_fut.n_max}
1069
+ - actor_ref_keybody_rel_pos_cur:
1070
+ func: ref_keybody_rel_pos_cur
1071
+ history_length: ${obs.context_length}
1072
+ flatten_history_dim: false
1073
+ params:
1074
+ ref_prefix: ${obs.actor_obs_prefix}
1075
+ keybody_names:
1076
+ - left_knee_link
1077
+ - right_knee_link
1078
+ - left_ankle_roll_link
1079
+ - right_ankle_roll_link
1080
+ - left_elbow_link
1081
+ - right_elbow_link
1082
+ - left_wrist_yaw_link
1083
+ - right_wrist_yaw_link
1084
+ noise:
1085
+ type: AdditiveUniformNoiseCfg
1086
+ params:
1087
+ n_min: ${domain_rand.obs_noise.actor_ref_keybody_rel_pos_cur.n_min}
1088
+ n_max: ${domain_rand.obs_noise.actor_ref_keybody_rel_pos_cur.n_max}
1089
+ - actor_ref_keybody_rel_pos_fut:
1090
+ func: ref_keybody_rel_pos_fut
1091
+ params:
1092
+ ref_prefix: ${obs.actor_obs_prefix}
1093
+ num_frames: ${obs.n_fut_frames}
1094
+ keybody_names:
1095
+ - left_knee_link
1096
+ - right_knee_link
1097
+ - left_ankle_roll_link
1098
+ - right_ankle_roll_link
1099
+ - left_elbow_link
1100
+ - right_elbow_link
1101
+ - left_wrist_yaw_link
1102
+ - right_wrist_yaw_link
1103
+ noise:
1104
+ type: AdditiveUniformNoiseCfg
1105
+ params:
1106
+ n_min: ${domain_rand.obs_noise.actor_ref_keybody_rel_pos_fut.n_min}
1107
+ n_max: ${domain_rand.obs_noise.actor_ref_keybody_rel_pos_fut.n_max}
1108
+ - actor_projected_gravity:
1109
+ func: projected_gravity
1110
+ history_length: ${obs.context_length}
1111
+ flatten_history_dim: false
1112
+ noise:
1113
+ type: AdditiveUniformNoiseCfg
1114
+ params:
1115
+ n_min: ${domain_rand.obs_noise.actor_projected_gravity.n_min}
1116
+ n_max: ${domain_rand.obs_noise.actor_projected_gravity.n_max}
1117
+ - actor_rel_robot_root_ang_vel:
1118
+ func: rel_robot_root_ang_vel
1119
+ history_length: ${obs.context_length}
1120
+ flatten_history_dim: false
1121
+ noise:
1122
+ type: AdditiveUniformNoiseCfg
1123
+ params:
1124
+ n_min: ${domain_rand.obs_noise.actor_rel_robot_root_ang_vel.n_min}
1125
+ n_max: ${domain_rand.obs_noise.actor_rel_robot_root_ang_vel.n_max}
1126
+ - actor_dof_pos:
1127
+ func: dof_pos
1128
+ history_length: ${obs.context_length}
1129
+ flatten_history_dim: false
1130
+ noise:
1131
+ type: AdditiveUniformNoiseCfg
1132
+ params:
1133
+ n_min: ${domain_rand.obs_noise.actor_dof_pos.n_min}
1134
+ n_max: ${domain_rand.obs_noise.actor_dof_pos.n_max}
1135
+ - actor_dof_vel:
1136
+ func: dof_vel
1137
+ history_length: ${obs.context_length}
1138
+ flatten_history_dim: false
1139
+ noise:
1140
+ type: AdditiveUniformNoiseCfg
1141
+ params:
1142
+ n_min: ${domain_rand.obs_noise.actor_dof_vel.n_min}
1143
+ n_max: ${domain_rand.obs_noise.actor_dof_vel.n_max}
1144
+ - actor_last_action:
1145
+ func: last_action
1146
+ history_length: ${obs.context_length}
1147
+ flatten_history_dim: false
1148
+ - critic_ref_dof_pos_cur:
1149
+ func: ref_dof_pos_cur
1150
+ params:
1151
+ ref_prefix: ${obs.critic_obs_prefix}
1152
+ - critic_ref_dof_pos_fut:
1153
+ func: ref_dof_pos_fut
1154
+ params:
1155
+ ref_prefix: ${obs.critic_obs_prefix}
1156
+ num_frames: ${obs.n_fut_frames}
1157
+ - critic_ref_root_height_fut:
1158
+ func: ref_root_height_fut
1159
+ params:
1160
+ ref_prefix: ${obs.critic_obs_prefix}
1161
+ num_frames: ${obs.n_fut_frames}
1162
+ - critic_ref_root_height_cur:
1163
+ func: ref_root_height_cur
1164
+ params:
1165
+ ref_prefix: ${obs.critic_obs_prefix}
1166
+ - critic_global_anchor_diff:
1167
+ func: global_anchor_diff
1168
+ params:
1169
+ ref_prefix: ${obs.critic_obs_prefix}
1170
+ - critic_ref_motion_cur_heading_aligned_root_pos:
1171
+ func: ref_motion_cur_heading_aligned_root_pos
1172
+ params:
1173
+ ref_prefix: ${obs.critic_obs_prefix}
1174
+ - critic_ref_motion_fut_heading_aligned_root_pos:
1175
+ func: ref_motion_fut_heading_aligned_root_pos
1176
+ params:
1177
+ ref_prefix: ${obs.critic_obs_prefix}
1178
+ num_frames: ${obs.n_fut_frames}
1179
+ - critic_ref_motion_cur_heading_aligned_root_rot6d:
1180
+ func: ref_motion_cur_heading_aligned_root_rot6d
1181
+ params:
1182
+ ref_prefix: ${obs.critic_obs_prefix}
1183
+ - critic_ref_motion_fut_heading_aligned_root_rot6d:
1184
+ func: ref_motion_fut_heading_aligned_root_rot6d
1185
+ params:
1186
+ ref_prefix: ${obs.critic_obs_prefix}
1187
+ num_frames: ${obs.n_fut_frames}
1188
+ - critic_ref_motion_cur_heading_aligned_root_lin_vel:
1189
+ func: ref_motion_cur_heading_aligned_root_lin_vel
1190
+ params:
1191
+ ref_prefix: ${obs.critic_obs_prefix}
1192
+ - critic_ref_motion_fut_heading_aligned_root_lin_vel:
1193
+ func: ref_motion_fut_heading_aligned_root_lin_vel
1194
+ params:
1195
+ ref_prefix: ${obs.critic_obs_prefix}
1196
+ num_frames: ${obs.n_fut_frames}
1197
+ - critic_ref_motion_cur_heading_aligned_root_ang_vel:
1198
+ func: ref_motion_cur_heading_aligned_root_ang_vel
1199
+ params:
1200
+ ref_prefix: ${obs.critic_obs_prefix}
1201
+ - critic_ref_motion_fut_heading_aligned_root_ang_vel:
1202
+ func: ref_motion_fut_heading_aligned_root_ang_vel
1203
+ params:
1204
+ ref_prefix: ${obs.critic_obs_prefix}
1205
+ num_frames: ${obs.n_fut_frames}
1206
+ - critic_rel_robot_root_lin_vel:
1207
+ func: rel_robot_root_lin_vel
1208
+ - critic_rel_robot_root_ang_vel:
1209
+ func: rel_robot_root_ang_vel
1210
+ - critic_global_robot_bodylink_lin_vel_flat:
1211
+ func: global_robot_bodylink_lin_vel_flat
1212
+ - critic_global_robot_bodylink_ang_vel_flat:
1213
+ func: global_robot_bodylink_ang_vel_flat
1214
+ - critic_root_rel_robot_bodylink_pos_flat:
1215
+ func: root_rel_robot_bodylink_pos_flat
1216
+ - critic_root_rel_robot_bodylink_rot_mat_flat:
1217
+ func: root_rel_robot_bodylink_rot_mat_flat
1218
+ - critic_dof_pos:
1219
+ func: dof_pos
1220
+ - critic_dof_vel:
1221
+ func: dof_vel
1222
+ - critic_last_action:
1223
+ func: last_action
1224
+ enable_corruption: true
1225
+ concatenate_terms: false
1226
+ additional_atomic_obs_list:
1227
+ - actor_ref_robot_yaw_error_sin_cos:
1228
+ func: ref_robot_yaw_error_sin_cos
1229
+ history_length: ${obs.context_length}
1230
+ flatten_history_dim: false
1231
+ params:
1232
+ ref_prefix: ${obs.actor_obs_prefix}
1233
+ - actor_ref_future_yaw_delta_sin_cos:
1234
+ func: ref_future_yaw_delta_sin_cos
1235
+ params:
1236
+ ref_prefix: ${obs.actor_obs_prefix}
1237
+ num_frames: ${obs.n_fut_frames}
1238
+ - actor_ref_future_root_ori_robot_frame_6d:
1239
+ func: ref_future_root_ori_robot_frame_6d
1240
+ params:
1241
+ ref_prefix: ${obs.actor_obs_prefix}
1242
+ num_frames: ${obs.n_fut_frames}
1243
+ rewards:
1244
+ _config:
1245
+ reward_prefix: ref_
1246
+ is_alive:
1247
+ weight: 0.5
1248
+ params: {}
1249
+ root_pos_xy_tracking_exp:
1250
+ weight: 0.0
1251
+ params:
1252
+ std: 0.2
1253
+ ref_prefix: ${rewards._config.reward_prefix}
1254
+ root_rot_tracking_exp:
1255
+ weight: 0.0
1256
+ params:
1257
+ std: 0.4
1258
+ ref_prefix: ${rewards._config.reward_prefix}
1259
+ root_rel_keybodylink_pos_tracking_l2_exp:
1260
+ weight: 1.0
1261
+ params:
1262
+ keybody_names: ${robot.key_bodies}
1263
+ std: 0.3
1264
+ ref_prefix: ${rewards._config.reward_prefix}
1265
+ root_rel_keybodylink_rot_tracking_l2_exp:
1266
+ weight: 1.0
1267
+ params:
1268
+ keybody_names: ${robot.key_bodies}
1269
+ std: 0.4
1270
+ ref_prefix: ${rewards._config.reward_prefix}
1271
+ root_lin_vel_tracking_rel_ratio_exp:
1272
+ weight: 1.0
1273
+ params:
1274
+ std: 1.0
1275
+ eps: 0.1
1276
+ ref_prefix: ${rewards._config.reward_prefix}
1277
+ root_ang_vel_tracking_rel_ratio_exp:
1278
+ weight: 1.0
1279
+ params:
1280
+ std: 1.0
1281
+ eps: 0.1
1282
+ ref_prefix: ${rewards._config.reward_prefix}
1283
+ local_reward_point_body_pos_tracking_exp:
1284
+ weight: 2.0
1285
+ params:
1286
+ std: 0.1
1287
+ ref_prefix: ${rewards._config.reward_prefix}
1288
+ reward_point_body:
1289
+ - torso_link
1290
+ - left_wrist_yaw_link
1291
+ - right_wrist_yaw_link
1292
+ - left_ankle_roll_link
1293
+ - right_ankle_roll_link
1294
+ reward_point_body_offset:
1295
+ - - 0.0
1296
+ - 0.0
1297
+ - 0.5
1298
+ - - 0.0
1299
+ - 0.0
1300
+ - 0.0
1301
+ - - 0.0
1302
+ - 0.0
1303
+ - 0.0
1304
+ - - 0.0
1305
+ - 0.0
1306
+ - 0.0
1307
+ - - 0.0
1308
+ - 0.0
1309
+ - 0.0
1310
+ point_weights:
1311
+ - 1.0
1312
+ - 1.0
1313
+ - 1.0
1314
+ - 1.0
1315
+ - 1.0
1316
+ global_keybodylink_lin_vel_tracking_l2_exp:
1317
+ weight: 1.0
1318
+ params:
1319
+ keybody_names: ${robot.key_bodies}
1320
+ std: 1.0
1321
+ ref_prefix: ${rewards._config.reward_prefix}
1322
+ global_keybodylink_ang_vel_tracking_l2_exp:
1323
+ weight: 1.0
1324
+ params:
1325
+ keybody_names: ${robot.key_bodies}
1326
+ std: 3.14
1327
+ ref_prefix: ${rewards._config.reward_prefix}
1328
+ action_rate_l2:
1329
+ weight: -0.1
1330
+ params: {}
1331
+ joint_pos_limits:
1332
+ weight: -10.0
1333
+ params:
1334
+ asset_cfg:
1335
+ _target_: isaaclab.managers.scene_entity_cfg.SceneEntityCfg
1336
+ name: robot
1337
+ joint_names:
1338
+ - .*
1339
+ undesired_contacts:
1340
+ weight: -0.1
1341
+ params:
1342
+ threshold: 1.0
1343
+ sensor_cfg:
1344
+ _target_: isaaclab.managers.scene_entity_cfg.SceneEntityCfg
1345
+ name: contact_forces
1346
+ body_names:
1347
+ - ${robot.undesired_contacts_regrex}
1348
+ domain_rand:
1349
+ action_delay:
1350
+ enabled: true
1351
+ min_delay: 0
1352
+ max_delay: 2
1353
+ erfi:
1354
+ enabled: false
1355
+ rfi_probability: 0.5
1356
+ rfi_lim: 0.1
1357
+ randomize_rfi_lim: true
1358
+ rfi_lim_range:
1359
+ - 0.5
1360
+ - 1.5
1361
+ rao_lim: 0.1
1362
+ obs_noise:
1363
+ actor_ref_gravity_projection_cur:
1364
+ n_min: -0.1
1365
+ n_max: 0.1
1366
+ actor_ref_gravity_projection_fut:
1367
+ n_min: -0.1
1368
+ n_max: 0.1
1369
+ actor_ref_base_linvel_cur:
1370
+ n_min: -0.1
1371
+ n_max: 0.1
1372
+ n_min_z: -0.05
1373
+ n_max_z: 0.05
1374
+ actor_ref_base_linvel_fut:
1375
+ n_min: -0.1
1376
+ n_max: 0.1
1377
+ n_min_z: -0.05
1378
+ n_max_z: 0.05
1379
+ actor_ref_base_angvel_cur:
1380
+ n_min: -0.1
1381
+ n_max: 0.1
1382
+ n_min_z: -0.1
1383
+ n_max_z: 0.1
1384
+ actor_ref_base_angvel_fut:
1385
+ n_min: -0.1
1386
+ n_max: 0.1
1387
+ n_min_z: -0.1
1388
+ n_max_z: 0.1
1389
+ actor_ref_dof_pos_cur:
1390
+ n_min: -0.05
1391
+ n_max: 0.05
1392
+ actor_ref_dof_pos_fut:
1393
+ n_min: -0.05
1394
+ n_max: 0.05
1395
+ actor_ref_root_height_cur:
1396
+ n_min: -0.1
1397
+ n_max: 0.1
1398
+ actor_ref_root_height_fut:
1399
+ n_min: -0.1
1400
+ n_max: 0.1
1401
+ actor_ref_keybody_rel_pos_cur:
1402
+ n_min: -0.1
1403
+ n_max: 0.1
1404
+ actor_ref_keybody_rel_pos_fut:
1405
+ n_min: -0.1
1406
+ n_max: 0.1
1407
+ actor_projected_gravity:
1408
+ n_min: -0.1
1409
+ n_max: 0.1
1410
+ actor_rel_robot_root_ang_vel:
1411
+ n_min: -0.2
1412
+ n_max: 0.2
1413
+ actor_dof_pos:
1414
+ n_min: -0.01
1415
+ n_max: 0.01
1416
+ actor_dof_vel:
1417
+ n_min: -0.5
1418
+ n_max: 0.5
1419
+ motion_init_perturb:
1420
+ root_pose_perturb_range:
1421
+ x:
1422
+ - -0.05
1423
+ - 0.05
1424
+ 'y':
1425
+ - -0.05
1426
+ - 0.05
1427
+ z:
1428
+ - -0.01
1429
+ - 0.01
1430
+ roll:
1431
+ - -0.1
1432
+ - 0.1
1433
+ pitch:
1434
+ - -0.1
1435
+ - 0.1
1436
+ yaw:
1437
+ - -0.2
1438
+ - 0.2
1439
+ root_vel_perturb_range:
1440
+ x:
1441
+ - -0.5
1442
+ - 0.5
1443
+ 'y':
1444
+ - -0.5
1445
+ - 0.5
1446
+ z:
1447
+ - -0.2
1448
+ - 0.2
1449
+ roll:
1450
+ - -0.5
1451
+ - 0.5
1452
+ pitch:
1453
+ - -0.5
1454
+ - 0.5
1455
+ yaw:
1456
+ - -0.2
1457
+ - 0.2
1458
+ dof_pos_perturb_range:
1459
+ - -0.1
1460
+ - 0.1
1461
+ dof_vel_perturb_range:
1462
+ - 0.0
1463
+ - 0.0
1464
+ default_dof_pos_bias:
1465
+ mode: startup
1466
+ params:
1467
+ joint_names:
1468
+ - .*
1469
+ pos_distribution_params:
1470
+ - -0.01
1471
+ - 0.01
1472
+ operation: add
1473
+ distribution: uniform
1474
+ rigid_body_com:
1475
+ mode: startup
1476
+ params:
1477
+ body_names: torso_link
1478
+ com_range:
1479
+ x:
1480
+ - -0.075
1481
+ - 0.075
1482
+ 'y':
1483
+ - -0.1
1484
+ - 0.1
1485
+ z:
1486
+ - -0.1
1487
+ - 0.1
1488
+ randomize_mass:
1489
+ mode: startup
1490
+ params:
1491
+ body_names:
1492
+ - pelvis
1493
+ - torso_link
1494
+ mass_range:
1495
+ - -1.0
1496
+ - 2.0
1497
+ rigid_body_material:
1498
+ mode: startup
1499
+ params:
1500
+ body_names: .*
1501
+ static_friction_range:
1502
+ - 0.3
1503
+ - 1.6
1504
+ dynamic_friction_range:
1505
+ - 0.3
1506
+ - 1.2
1507
+ restitution_range:
1508
+ - 0.0
1509
+ - 0.5
1510
+ num_buckets: 64
1511
+ push_by_setting_velocity:
1512
+ mode: interval
1513
+ interval_range_s:
1514
+ - 1.0
1515
+ - 3.0
1516
+ params:
1517
+ velocity_range:
1518
+ x:
1519
+ - -0.5
1520
+ - 0.5
1521
+ 'y':
1522
+ - -0.5
1523
+ - 0.5
1524
+ z:
1525
+ - -0.2
1526
+ - 0.2
1527
+ roll:
1528
+ - -0.52
1529
+ - 0.52
1530
+ pitch:
1531
+ - -0.52
1532
+ - 0.52
1533
+ yaw:
1534
+ - -0.78
1535
+ - 0.78
1536
+ randomize_actuator_gains:
1537
+ mode: startup
1538
+ params:
1539
+ asset_name: robot
1540
+ body_names: .*
1541
+ stiffness_distribution_params:
1542
+ - 0.9
1543
+ - 1.1
1544
+ damping_distribution_params:
1545
+ - 0.9
1546
+ - 1.1
1547
+ operation: scale
1548
+ distribution: uniform
1549
+ terrain:
1550
+ terrain_type: generator
1551
+ prim_path: /World/ground
1552
+ static_friction: 1.0
1553
+ dynamic_friction: 1.0
1554
+ restitution: 0.0
1555
+ friction_combine_mode: multiply
1556
+ restitution_combine_mode: multiply
1557
+ debug_vis: false
1558
+ max_init_terrain_level: 4
1559
+ random_spawn: true
1560
+ random_spawn_margin: 4.0
1561
+ generator:
1562
+ num_rows: 4
1563
+ num_cols: 4
1564
+ size:
1565
+ - 20.0
1566
+ - 20.0
1567
+ border_width: 1000.0
1568
+ horizontal_scale: 0.1
1569
+ vertical_scale: 0.005
1570
+ slope_threshold: null
1571
+ difficulty_range:
1572
+ - 0.0
1573
+ - 1.0
1574
+ color_scheme: height
1575
+ sub_terrains:
1576
+ rough:
1577
+ type: random_uniform
1578
+ proportion: 1.0
1579
+ noise_range:
1580
+ - 0.0
1581
+ - 0.04
1582
+ noise_step: 0.05
1583
+ downsampled_scale: 1.0
1584
+ visual_material:
1585
+ type: color
1586
+ diffuse_color:
1587
+ - 0.25
1588
+ - 0.25
1589
+ - 0.25
1590
+ metallic: 0.0
1591
+ roughness: 0.5
1592
+ modules:
1593
+ actor:
1594
+ type: ReferenceRoutedGroupedMoETransformerPolicy
1595
+ use_checkpointing: false
1596
+ num_fine_experts: 1024
1597
+ num_shared_experts: 1
1598
+ top_k: 2
1599
+ moe_loss_coef: 0.0
1600
+ routing_score_fn: ${algo.config.moe_router.routing_score_fn}
1601
+ routing_scale: ${algo.config.moe_router.routing_scale}
1602
+ use_dynamic_bias: ${algo.config.moe_router.use_dynamic_bias}
1603
+ bias_update_rate: ${algo.config.moe_router.bias_update_rate}
1604
+ expert_bias_clip: ${algo.config.moe_router.expert_bias_clip}
1605
+ obs_embed_mlp_hidden: 2048
1606
+ router_embed_mlp_hidden: 2048
1607
+ d_model: 512
1608
+ n_heads: 8
1609
+ n_kv_heads: 4
1610
+ use_gated_attn: true
1611
+ n_layers: 1
1612
+ dense_layer_at_first: false
1613
+ ff_mult: 0.5
1614
+ ff_mult_dense: 4
1615
+ attn_dropout: 0.0
1616
+ mlp_dropout: 0.0
1617
+ max_ctx_len: 32
1618
+ aux_sys_id_weight: 0.0
1619
+ aux_dynamics_weight: 0.0
1620
+ obs_norm:
1621
+ enabled: true
1622
+ epsilon: 1.0e-08
1623
+ update_method: ema
1624
+ ema_momentum: 0.0001
1625
+ update_at_train: true
1626
+ update_at_eval: false
1627
+ enable_clipping: true
1628
+ clip_range: 10.0
1629
+ sync_interval_steps: 4
1630
+ obs_schema:
1631
+ flattened_obs:
1632
+ seq_len: ${obs.context_length}
1633
+ terms:
1634
+ - unified/actor_ref_gravity_projection_cur
1635
+ - unified/actor_ref_base_linvel_cur
1636
+ - unified/actor_ref_base_angvel_cur
1637
+ - unified/actor_ref_dof_pos_cur
1638
+ - unified/actor_ref_root_height_cur
1639
+ - unified/actor_ref_robot_yaw_error_sin_cos
1640
+ - unified/actor_projected_gravity
1641
+ - unified/actor_rel_robot_root_ang_vel
1642
+ - unified/actor_dof_pos
1643
+ - unified/actor_dof_vel
1644
+ - unified/actor_last_action
1645
+ flattened_obs_fut:
1646
+ seq_len: ${obs.n_fut_frames}
1647
+ terms:
1648
+ - unified/actor_ref_dof_pos_fut
1649
+ - unified/actor_ref_root_height_fut
1650
+ - unified/actor_ref_gravity_projection_fut
1651
+ - unified/actor_ref_base_linvel_fut
1652
+ - unified/actor_ref_base_angvel_fut
1653
+ - unified/actor_ref_future_yaw_delta_sin_cos
1654
+ - unified/actor_ref_future_root_ori_robot_frame_6d
1655
+ output_dim: robot_action_dim
1656
+ critic:
1657
+ type: MLP
1658
+ obs_norm:
1659
+ enabled: true
1660
+ epsilon: 1.0e-08
1661
+ update_method: ema
1662
+ ema_momentum: 0.0001
1663
+ update_at_train: true
1664
+ update_at_eval: false
1665
+ enable_clipping: true
1666
+ clip_range: 10.0
1667
+ sync_interval_steps: 4
1668
+ hidden_norm: rmsnorm
1669
+ layer_config:
1670
+ hidden_dims:
1671
+ - 2048
1672
+ - 2048
1673
+ - 2048
1674
+ - 2048
1675
+ activation: SiLU
1676
+ obs_schema:
1677
+ flattened_obs:
1678
+ seq_len: 1
1679
+ terms:
1680
+ - unified/critic_ref_dof_pos_cur
1681
+ - unified/critic_global_anchor_diff
1682
+ - unified/critic_ref_motion_cur_heading_aligned_root_pos
1683
+ - unified/critic_ref_motion_cur_heading_aligned_root_rot6d
1684
+ - unified/critic_ref_motion_cur_heading_aligned_root_lin_vel
1685
+ - unified/critic_ref_motion_cur_heading_aligned_root_ang_vel
1686
+ - unified/critic_rel_robot_root_lin_vel
1687
+ - unified/critic_rel_robot_root_ang_vel
1688
+ - unified/critic_global_robot_bodylink_lin_vel_flat
1689
+ - unified/critic_global_robot_bodylink_ang_vel_flat
1690
+ - unified/critic_root_rel_robot_bodylink_pos_flat
1691
+ - unified/critic_root_rel_robot_bodylink_rot_mat_flat
1692
+ - unified/critic_dof_pos
1693
+ - unified/critic_dof_vel
1694
+ - unified/critic_last_action
1695
+ flattened_obs_fut:
1696
+ seq_len: ${obs.n_fut_frames}
1697
+ terms:
1698
+ - unified/critic_ref_dof_pos_fut
1699
+ - unified/critic_ref_root_height_fut
1700
+ - unified/critic_ref_motion_fut_heading_aligned_root_pos
1701
+ - unified/critic_ref_motion_fut_heading_aligned_root_rot6d
1702
+ - unified/critic_ref_motion_fut_heading_aligned_root_lin_vel
1703
+ - unified/critic_ref_motion_fut_heading_aligned_root_ang_vel
1704
+ output_dim: 1
1705
+ train_hdf5_roots:
1706
+ - /horizon-bucket/robot_lab/users/bo07.zhang-labs/Data/h5v2_datasets/h5v2_AMASS_GMR
1707
+ - /horizon-bucket/robot_lab/users/bo07.zhang-labs/Data/h5v2_datasets/h5v2_LAFAN1_g1_official
1708
+ - /horizon-bucket/robot_lab/users/bo07.zhang-labs/Data/h5v2_datasets/h5v2_pico_record_260402
1709
+ - /horizon-bucket/robot_lab/users/bo07.zhang-labs/Data/h5v2_datasets/h5v2_MotionMillion_GMR_260402
1710
+ - /horizon-bucket/robot_lab/users/bo07.zhang-labs/Data/h5v2_datasets/h5v2_nyt_260410
1711
+ - /horizon-bucket/robot_lab/users/bo07.zhang-labs/Data/h5v2_datasets/h5v2_sonic_260430
1712
+ mujoco_eval:
1713
+ enabled: true
1714
+ robot_xml_path: assets/robots/unitree/G1/29dof/scene_29dof.xml
1715
+ motion_npz_dir: /horizon-bucket/robot_lab/users/maiyue01.chen/expr_output/HoloMotionMotionTracking/HoloMotionMotrackV1.2/20260216_234025-training/motion_tracking/train_g1_test_h5v2_tf_aidi_8-64_2500h_more_info_beta2_2x8GPUs/isaaclab_eval_output_model_15000_v1.1_eval
1716
+ dump_npzs: true
1717
+ calc_per_clip_metrics: true
1718
+ generate_report: true
1719
+ ray_actors_per_gpu: 16
1720
+ ray_multi_ckpt_mode: split
1721
+ ckpt_onnx_names:
1722
+ - model_2000.onnx
1723
+ - model_4000.onnx
1724
+ - model_6000.onnx
1725
+ - model_8000.onnx
1726
+ - model_10000.onnx
HoloMotion_motion_tracking_model_v1.3.2/exported/motion_tracking_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f3fa8cecf98080fe255348fd94627038f90db64246ca7309de15e80ad6bc6f22
3
+ size 1640360076