Caesarrr commited on
Commit
70b5f5a
·
verified ·
1 Parent(s): ec447a7

Bind hopper Pi0.5 legacy state and actions to the current inference code

Browse files
latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/config.yaml CHANGED
@@ -102,7 +102,7 @@ datasets:
102
  - leg_angular_velocity
103
  - foot_angular_velocity
104
  task_name: hopper
105
- robot_type: rl_games_gymnasium
106
  trainer:
107
  pretrained_checkpoint: null
108
  profile_timing:
 
102
  - leg_angular_velocity
103
  - foot_angular_velocity
104
  task_name: hopper
105
+ robot_type: rl_games_gymnasium_native
106
  trainer:
107
  pretrained_checkpoint: null
108
  profile_timing:
latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/dataset_statistics.json CHANGED
@@ -22,12 +22,12 @@
22
  -1.0
23
  ],
24
  "q01": [
25
- -0.5339896190166473,
26
  -1.0,
27
  -1.0
28
  ],
29
  "q99": [
30
- 0.6878953921794876,
31
  1.0,
32
  1.0
33
  ],
@@ -91,29 +91,29 @@
91
  -10.0
92
  ],
93
  "q01": [
94
- 1.0245488870695927,
95
- -0.15144138299100837,
96
- -0.7427855337859313,
97
- -0.8104351850643912,
98
- -0.8960894427439994,
99
- 0.44528900369002233,
100
- -2.74102122555904,
101
- -2.6237940408222125,
102
- -4.557577227105824,
103
- -9.73546984195709,
104
  -10.0
105
  ],
106
  "q99": [
107
- 1.746462032848763,
108
- 0.07944139343900171,
109
- -0.009743861129406839,
110
- 0.06964989517986797,
111
- 0.9054187285678829,
112
- 3.9090380648781706,
113
- 2.899137949682023,
114
- 2.2594460707147905,
115
- 5.696335758252259,
116
- 9.774792885780334,
117
  10.0
118
  ]
119
  },
 
22
  -1.0
23
  ],
24
  "q01": [
25
+ -1.0,
26
  -1.0,
27
  -1.0
28
  ],
29
  "q99": [
30
+ 1.0,
31
  1.0,
32
  1.0
33
  ],
 
91
  -10.0
92
  ],
93
  "q01": [
94
+ 0.701430082321167,
95
+ -0.18769849836826324,
96
+ -1.4578222036361694,
97
+ -1.9729520082473755,
98
+ -0.9319470524787903,
99
+ -0.014216306619346142,
100
+ -3.833221197128296,
101
+ -6.138384819030762,
102
+ -10.0,
103
+ -10.0,
104
  -10.0
105
  ],
106
  "q99": [
107
+ 1.8196797370910645,
108
+ 0.1958828866481781,
109
+ 0.041033145040273666,
110
+ 0.1550283581018448,
111
+ 0.9662267565727234,
112
+ 7.04247522354126,
113
+ 3.2499496936798096,
114
+ 5.976532459259033,
115
+ 7.204126834869385,
116
+ 10.0,
117
  10.0
118
  ]
119
  },
latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/manifest.json CHANGED
@@ -72,7 +72,7 @@
72
  10.0
73
  ]
74
  },
75
- "robot_type": "rl_games_gymnasium",
76
  "latency_prompt_map_path": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/lerobot/latency_prompt_map.json",
77
  "custom_mixtures_path": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/_generated_mixtures/lerobot.json",
78
  "gymnasium_task": {
@@ -140,4 +140,4 @@
140
  "validation_dataset_name": "lerobot__val",
141
  "validation_episodes": 10,
142
  "validation_frames": 10000
143
- }
 
72
  10.0
73
  ]
74
  },
75
+ "robot_type": "rl_games_gymnasium_native",
76
  "latency_prompt_map_path": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/lerobot/latency_prompt_map.json",
77
  "custom_mixtures_path": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/_generated_mixtures/lerobot.json",
78
  "gymnasium_task": {
 
140
  "validation_dataset_name": "lerobot__val",
141
  "validation_episodes": 10,
142
  "validation_frames": 10000
143
+ }
latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/normalization_binding.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "weights": "unchanged",
3
  "original_statistics": "original_training_statistics.json",
4
- "state_transform": "Raw-state min/max from the source manifest reproduces converter-normalized training inputs.",
5
  "manifest": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/new_bundle/manifest.json"
6
  }
 
1
  {
2
  "weights": "unchanged",
3
  "original_statistics": "original_training_statistics.json",
4
+ "state_transform": "State statistics reverse source converter coordinates; the legacy Gymnasium DataConfig uses clipped min/max state and identity native actions.",
5
  "manifest": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/new_bundle/manifest.json"
6
  }
latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/source_manifest.json ADDED
@@ -0,0 +1,143 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_name": "lerobot",
3
+ "env_name": "hopper",
4
+ "episodes": 90,
5
+ "frames": 87583,
6
+ "task_prompts": [
7
+ "Move the Hopper robot forward while keeping its torso upright. Predict three continuous torques in [-1, 1] ordered as thigh, leg, and foot. Current action latency is 1 raw frames (200.00 ms). The environment runs at 5 FPS and observations are emitted at 5 FPS. Choose the best next action.",
8
+ "Move the Hopper robot forward while keeping its torso upright. Predict three continuous torques in [-1, 1] ordered as thigh, leg, and foot. Current action latency is 2 raw frames (400.00 ms). The environment runs at 5 FPS and observations are emitted at 5 FPS. Choose the best next action."
9
+ ],
10
+ "format": "lerobot_v2_image_parquet",
11
+ "source": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/raw",
12
+ "integration_name": "gymnasium",
13
+ "task_name": "hopper",
14
+ "action_layout": "gymnasium_continuous_v1",
15
+ "action_labels": [
16
+ "thigh_torque",
17
+ "leg_torque",
18
+ "foot_torque"
19
+ ],
20
+ "carrier_action_labels": [
21
+ "thigh_torque",
22
+ "leg_torque",
23
+ "foot_torque"
24
+ ],
25
+ "action_dim": 3,
26
+ "active_action_dim": 3,
27
+ "action_carrier": "native",
28
+ "rows_unit": "decision_step",
29
+ "obs_fps": 5.0,
30
+ "obs_stride_raw_frames": 1,
31
+ "uses_state": true,
32
+ "state_dim": 11,
33
+ "state_labels": [
34
+ "torso_height",
35
+ "torso_angle",
36
+ "thigh_angle",
37
+ "leg_angle",
38
+ "foot_angle",
39
+ "torso_x_velocity",
40
+ "torso_z_velocity",
41
+ "torso_angular_velocity",
42
+ "thigh_angular_velocity",
43
+ "leg_angular_velocity",
44
+ "foot_angular_velocity"
45
+ ],
46
+ "state_normalization": {
47
+ "type": "min_max",
48
+ "min": [
49
+ 0.701430082321167,
50
+ -0.18769849836826324,
51
+ -1.4578222036361694,
52
+ -1.9729520082473755,
53
+ -0.9319470524787903,
54
+ -0.014216306619346142,
55
+ -3.833221197128296,
56
+ -6.138384819030762,
57
+ -10.0,
58
+ -10.0,
59
+ -10.0
60
+ ],
61
+ "max": [
62
+ 1.8196797370910645,
63
+ 0.1958828866481781,
64
+ 0.041033145040273666,
65
+ 0.1550283581018448,
66
+ 0.9662267565727234,
67
+ 7.04247522354126,
68
+ 3.2499496936798096,
69
+ 5.976532459259033,
70
+ 7.204126834869385,
71
+ 10.0,
72
+ 10.0
73
+ ]
74
+ },
75
+ "robot_type": "rl_games_gymnasium",
76
+ "latency_prompt_map_path": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/lerobot/latency_prompt_map.json",
77
+ "custom_mixtures_path": "/home/ubuntu/lzj/latency-sensitive-bench/outputs/pi05_issued_action_iid_s42_20261010/hopper/_generated_mixtures/lerobot.json",
78
+ "gymnasium_task": {
79
+ "action_space": {
80
+ "dtype": "float32",
81
+ "high": [
82
+ 1.0,
83
+ 1.0,
84
+ 1.0
85
+ ],
86
+ "labels": [
87
+ "thigh_torque",
88
+ "leg_torque",
89
+ "foot_torque"
90
+ ],
91
+ "low": [
92
+ -1.0,
93
+ -1.0,
94
+ -1.0
95
+ ],
96
+ "type": "box"
97
+ },
98
+ "base_prompt": "Move the Hopper robot forward while keeping its torso upright. Predict three continuous torques in [-1, 1] ordered as thigh, leg, and foot.",
99
+ "env_fps": 5.0,
100
+ "env_id": "LatencyBench/Hopper-v0",
101
+ "frame_stack": 1,
102
+ "make_kwargs": {
103
+ "base_env_id": "Hopper-v4",
104
+ "base_make_kwargs": {
105
+ "ctrl_cost_weight": 0.001,
106
+ "exclude_current_positions_from_observation": true,
107
+ "forward_reward_weight": 1.0,
108
+ "healthy_reward": 1.0,
109
+ "reset_noise_scale": 0.005,
110
+ "terminate_when_unhealthy": true
111
+ },
112
+ "render_mode": "rgb_array"
113
+ },
114
+ "noop_action": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0
118
+ ],
119
+ "obs_fps": 5.0,
120
+ "registration_imports": [
121
+ "latency_bench.envs.tasks.hopper"
122
+ ],
123
+ "state_space": {
124
+ "labels": [
125
+ "torso_height",
126
+ "torso_angle",
127
+ "thigh_angle",
128
+ "leg_angle",
129
+ "foot_angle",
130
+ "torso_x_velocity",
131
+ "torso_z_velocity",
132
+ "torso_angular_velocity",
133
+ "thigh_angular_velocity",
134
+ "leg_angular_velocity",
135
+ "foot_angular_velocity"
136
+ ]
137
+ },
138
+ "task_name": "hopper"
139
+ },
140
+ "validation_dataset_name": "lerobot__val",
141
+ "validation_episodes": 10,
142
+ "validation_frames": 10000
143
+ }