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 +1 -1
- latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/dataset_statistics.json +22 -22
- latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/manifest.json +2 -2
- latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/normalization_binding.json +1 -1
- latency-aware/hopper/vla/starvla-qwenpi_v3-h1/hopper_pi05_profile_gt2500_h1_5fps_g128_20260922/source_manifest.json +143 -0
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:
|
| 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 |
-
-
|
| 26 |
-1.0,
|
| 27 |
-1.0
|
| 28 |
],
|
| 29 |
"q99": [
|
| 30 |
-
|
| 31 |
1.0,
|
| 32 |
1.0
|
| 33 |
],
|
|
@@ -91,29 +91,29 @@
|
|
| 91 |
-10.0
|
| 92 |
],
|
| 93 |
"q01": [
|
| 94 |
-
|
| 95 |
-
-0.
|
| 96 |
-
-
|
| 97 |
-
-
|
| 98 |
-
-0.
|
| 99 |
-
0.
|
| 100 |
-
-
|
| 101 |
-
-
|
| 102 |
-
-
|
| 103 |
-
-
|
| 104 |
-10.0
|
| 105 |
],
|
| 106 |
"q99": [
|
| 107 |
-
1.
|
| 108 |
-
0.
|
| 109 |
-
|
| 110 |
-
0.
|
| 111 |
-
0.
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
-
|
| 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": "
|
| 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": "
|
| 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 |
+
}
|