feat: add 30s dance (dance1_subject2) longer-horizon DeepMimic PPO ckpt
Browse files- dance_30s/env.yaml +41 -0
- dance_30s/model.pt +3 -0
- dance_30s/motion.pkl +3 -0
dance_30s/env.yaml
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
env_name: "deepmimic"
|
| 2 |
+
|
| 3 |
+
char_file: "data/assets/g1/g1.xml"
|
| 4 |
+
camera_mode: "track"
|
| 5 |
+
|
| 6 |
+
episode_length: 10.0 # seconds
|
| 7 |
+
global_obs: True
|
| 8 |
+
root_height_obs: True
|
| 9 |
+
pose_termination: True
|
| 10 |
+
pose_termination_dist: 1.0 # m
|
| 11 |
+
enable_phase_obs: False
|
| 12 |
+
enable_tar_obs: True
|
| 13 |
+
num_phase_encoding: 4
|
| 14 |
+
tar_obs_steps: [1, 2, 3]
|
| 15 |
+
rand_reset: True
|
| 16 |
+
zero_center_action: True
|
| 17 |
+
log_tracking_error: False
|
| 18 |
+
|
| 19 |
+
ref_char_offset: [2.0, 0.0, 0.0] # m
|
| 20 |
+
init_pose: [0, 0, 0.8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1.57, 0, 0, 0, 0, 0, 0, 1.57, 0, 0, 0]
|
| 21 |
+
|
| 22 |
+
enable_early_termination: True
|
| 23 |
+
key_bodies: ["left_ankle_roll_link", "right_ankle_roll_link", "head_link", "left_wrist_yaw_link", "right_wrist_yaw_link"]
|
| 24 |
+
contact_bodies: ["left_knee_link", "left_ankle_pitch_link", "left_ankle_roll_link", "right_knee_link", "right_ankle_pitch_link", "right_ankle_roll_link"]
|
| 25 |
+
#contact_bodies: ["left_knee_link", "left_ankle_pitch_link", "left_ankle_roll_link", "right_knee_link", "right_ankle_pitch_link", "right_ankle_roll_link", "left_elbow_link", "left_wrist_roll_link", "left_wrist_pitch_link", "left_wrist_yaw_link", "right_elbow_link", "right_wrist_roll_link", "right_wrist_pitch_link", "right_wrist_yaw_link"] # cartwheel
|
| 26 |
+
|
| 27 |
+
motion_file: "data/motions/g1/lafan_dance_30s.pkl"
|
| 28 |
+
|
| 29 |
+
joint_err_w: [1.0, 1.0, 1.0, 0.6, 0.5, 0.5, 1.0, 1.0, 1.0, 0.6, 0.5, 0.5, 1.0, 1.0, 1.0, 0.0, 1.0, 1.0, 1.0, 0.6, 0.5, 0.5, 0.5, 1.0, 1.0, 1.0, 0.6, 0.5, 0.5, 0.5]
|
| 30 |
+
|
| 31 |
+
reward_pose_w: 0.5
|
| 32 |
+
reward_vel_w: 0.1
|
| 33 |
+
reward_root_pose_w: 0.15
|
| 34 |
+
reward_root_vel_w: 0.1
|
| 35 |
+
reward_key_pos_w: 0.15
|
| 36 |
+
|
| 37 |
+
reward_pose_scale: 0.25
|
| 38 |
+
reward_vel_scale: 0.01
|
| 39 |
+
reward_root_pose_scale: 5.0
|
| 40 |
+
reward_root_vel_scale: 1.0
|
| 41 |
+
reward_key_pos_scale: 10.0
|
dance_30s/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:91c8d558bb7c5a0e8c8d1b020a60ed6c03563688ad432b867c37ec0e6946240a
|
| 3 |
+
size 11236728
|
dance_30s/motion.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f16b730b442703af5ff85f094e937152b37f90cea9c83858c593888bfe3a84f5
|
| 3 |
+
size 287187
|