HoloMotion
ONNX
Safetensors
kaihui-hug123 commited on
Commit
b6bf0cd
·
verified ·
1 Parent(s): c2cd6bc

upload v1.4.0 motion tracking model

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