wsagi commited on
Commit
9b3c84e
·
verified ·
1 Parent(s): 24658bb

feat: add 30s dance (dance1_subject2) longer-horizon DeepMimic PPO ckpt

Browse files
Files changed (3) hide show
  1. dance_30s/env.yaml +41 -0
  2. dance_30s/model.pt +3 -0
  3. 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