{ "suite": "level2_rigid", "suite_id": "l2rigid", "description": "all level-2 rigid-body tasks", "campaign": "pi05_lv2sub_20261029", "tasks": [ "HangMug-v1", "PivotBook-v1", "OpenBoxRetrieval-v1", "KitAssembly-v1", "GraspTwoBlocks-v1", "ReorientPen-v1", "DexOpenDoor-v1", "HammerNailOut-v1", "DualArmLiftBall-v1", "DualArmBookPickPlace-v1", "DualArmPegInHole-v1", "DualArmCleanupPotEnv-v1", "DualArmOpenLaptop-v1", "DualArmThrowBlock-v1", "DualArmJugglingBall-v1", "DualArmOpenBottle-v1" ], "method": "PI0.5 LoRA + trainable weights, multi-task (one policy for all tasks in the suite)", "base_checkpoint": "gs://openpi-assets/checkpoints/pi05_base/params", "checkpoints": [ { "file": "lora_step_0010000.npz", "step": 10000, "sha256": "75f45669d2a8ff5a06ee65ddff8a813446b6001e3cf810a1226efba556f8352e", "bytes": 1867943218, "evaluated": false }, { "file": "lora_step_0020000.npz", "step": 20000, "sha256": "58ec22b36365e34305a0bba6447bcb8516182ec39573d92de6e3aefbaf71d1bc", "bytes": 1867943218, "evaluated": false }, { "file": "lora_step_0030000.npz", "step": 30000, "sha256": "43f4fd05650a42c5d2061ef69682e76168d0ddb0a1080c9e223111aee6c7cbf1", "bytes": 1867943218, "evaluated": false }, { "file": "lora_step_0040000.npz", "step": 40000, "sha256": "04de3049f6361a4f0ba9f2dcf52c76ca1aa1929d560fe80d9f758e914bb62973", "bytes": 1867943218, "evaluated": false }, { "file": "lora_step_0050000.npz", "step": 50000, "sha256": "51fd5f209443986f40ff7168b3ebde67465ffc8ce9f4b0b763f48f1be5534ec6", "bytes": 1867943218, "evaluated": true } ], "evaluated_checkpoint": "lora_step_0050000.npz", "tensor_count": 51, "action_dim": 44, "state_dim": 44, "action_dim_note": "44 = union of all level-2 embodiments (dual-arm dexterous); narrower embodiments are zero-padded, as in the level-2 dataset", "action_horizon": 50, "max_token_len": 200, "image_size": 224, "exec_steps": 16, "sample_steps": 10, "control_mode": "pd_ee_delta_pose", "state_input": "zero vector", "cameras": [ "front_camera", "robot_camera", "robot_camera_0", "robot_camera_1" ], "missing_right_wrist_image": "zeros", "num_demos_per_task": 20, "train_episodes": 320, "train_frames": 142831, "train_steps": 50000, "batch_size": 64, "lr": 5e-05, "warmup_steps": 1000, "lr_schedule": "cosine decay over all 50000 steps to 5e-6", "data_parallel": "pure data parallel (fsdp_devices=1), 2-8 GPUs, global batch 64", "norm_stats_file": "norm_stats.json", "norm_stats_sha256": "014a5ba1aeac06ab3063bd1a4da191302fa7d71ea59ac544f0783e624c0a14d4", "norm_stats_note": "quantile stats computed on this suite's frames only; padded dims whose q01==q99 were repaired (padding-aware)", "training_data": "Linslab/ManiLadder_Dataset raw/, 20 demos per task (16 tasks, 320 episodes, 142831 frames)", "task_sources": { "HangMug-v1": { "path": "maniladder/envs/rigid_body/single_arm_gripper/level_2/hang_mug.py", "sha256": "18a95ccc658e8725d4d38a6e8f445bb8a22e6671b5647e0e49b87b14a2109f31", "level": "level_2", "category": "rigid_body/single_arm_gripper", "cameras": [ "front_camera", "robot_camera", null ] }, "PivotBook-v1": { "path": "maniladder/envs/rigid_body/single_arm_gripper/level_2/pivot_book.py", "sha256": "b4a162f84c4c43180ea4bc6e8a98326713273b862b4e24b5f8b03d6d6c8d93cc", "level": "level_2", "category": "rigid_body/single_arm_gripper", "cameras": [ "front_camera", "robot_camera", null ] }, "OpenBoxRetrieval-v1": { "path": "maniladder/envs/rigid_body/single_arm_gripper/level_2/open_box_retrieval.py", "sha256": "2fade45048b317313439c6f85cc569b0baeb16165f7457379c823abf63bed27b", "level": "level_2", "category": "rigid_body/single_arm_gripper", "cameras": [ "front_camera", "robot_camera", null ] }, "KitAssembly-v1": { "path": "maniladder/envs/rigid_body/single_arm_gripper/level_2/kit_assembly.py", "sha256": "d7161b09b2a3612f099cf4281e99103afe66f65a39b25dc4f24b9a6d97ed9b2a", "level": "level_2", "category": "rigid_body/single_arm_gripper", "cameras": [ "front_camera", "robot_camera", null ] }, "GraspTwoBlocks-v1": { "path": "maniladder/envs/rigid_body/single_arm_dex/level_2/grasp_blocks.py", "sha256": "5ca6052059d89129ea2053033b959eafe590e119664bbcf431143d371650d66e", "level": "level_2", "category": "rigid_body/single_arm_dex", "cameras": [ "front_camera", "robot_camera", null ] }, "ReorientPen-v1": { "path": "maniladder/envs/rigid_body/single_arm_dex/level_2/reorient_pen.py", "sha256": "cb46bbdf5dc53cc310e613a1556720d34b444b1333421682cd9b5133972aa386", "level": "level_2", "category": "rigid_body/single_arm_dex", "cameras": [ "front_camera", "robot_camera", null ] }, "DexOpenDoor-v1": { "path": "maniladder/envs/rigid_body/single_arm_dex/level_2/open_door.py", "sha256": "6c21f64c2a98c359f1d01c67743cd590799b763ba3e1b933b7204d798312d062", "level": "level_2", "category": "rigid_body/single_arm_dex", "cameras": [ "front_camera", "robot_camera", null ] }, "HammerNailOut-v1": { "path": "maniladder/envs/rigid_body/single_arm_dex/level_2/hammer_nail_out.py", "sha256": "f524116a35ee681f0573ed934a47914d71d1ab9ca9cfd610f7d43a574cb8bac4", "level": "level_2", "category": "rigid_body/single_arm_dex", "cameras": [ "front_camera", "robot_camera", null ] }, "DualArmLiftBall-v1": { "path": "maniladder/envs/rigid_body/dual_arm_gripper/level_2/lift_ball.py", "sha256": "0560e10a0e9d98906408f8171268cd04ce655a626f38850bca0f31bf04bd9154", "level": "level_2", "category": "rigid_body/dual_arm_gripper", "cameras": [ "front_camera", "robot_camera_0", "robot_camera_1" ] }, "DualArmBookPickPlace-v1": { "path": "maniladder/envs/rigid_body/dual_arm_gripper/level_2/book_pick_place.py", "sha256": "6a92d5e17c749bacb4dbb9c0aa8b5a6b3922710a70247473846b807e71e8e7a4", "level": "level_2", "category": "rigid_body/dual_arm_gripper", "cameras": [ "front_camera", "robot_camera_0", "robot_camera_1" ] }, "DualArmPegInHole-v1": { "path": "maniladder/envs/rigid_body/dual_arm_gripper/level_2/peg_in_hole.py", "sha256": "38c5217e32fae22a0fa962b54c44a42fc12cd1220060d602365b8be75f42e5ae", "level": "level_2", "category": "rigid_body/dual_arm_gripper", "cameras": [ "front_camera", "robot_camera_0", "robot_camera_1" ] }, "DualArmCleanupPotEnv-v1": { "path": "maniladder/envs/rigid_body/dual_arm_gripper/level_2/cleanup_pot.py", "sha256": "279f544967eb0539e28bc8cd4063d28b80aa0fead7c0733a1787ad0fc983b155", "level": "level_2", "category": "rigid_body/dual_arm_gripper", "cameras": [ "front_camera", "robot_camera_0", "robot_camera_1" ] }, "DualArmOpenLaptop-v1": { "path": "maniladder/envs/rigid_body/dual_arm_dex/level_2/open_laptop.py", "sha256": "e2c3c0714da9a484ed7a7ac3dc6374f61c7e1867167f50b37dca9845534faf93", "level": "level_2", "category": "rigid_body/dual_arm_dex", "cameras": [ "front_camera", "robot_camera_0", "robot_camera_1" ] }, "DualArmThrowBlock-v1": { "path": "maniladder/envs/rigid_body/dual_arm_dex/level_2/throw_ball.py", "sha256": "0a00a7e724272b98eb42cd58fb57f6e15756a6e115f19f320febda6dda6d7680", "level": "level_2", "category": "rigid_body/dual_arm_dex", "cameras": [ "front_camera", "robot_camera_0", "robot_camera_1" ] }, "DualArmJugglingBall-v1": { "path": "maniladder/envs/rigid_body/dual_arm_dex/level_2/juggle_ball.py", "sha256": "f5a65ccfa2d7aa379f131628849315ce1840b5e11620e5f6a86c6a3451706ed1", "level": "level_2", "category": "rigid_body/dual_arm_dex", "cameras": [ "front_camera", "robot_camera_0", "robot_camera_1" ] }, "DualArmOpenBottle-v1": { "path": "maniladder/envs/rigid_body/dual_arm_dex/level_2/open_bottle.py", "sha256": "6c492aff17adfba725a0f605df5394f4fe93e04ef122cf473a210e4a2cf84e2f", "level": "level_2", "category": "rigid_body/dual_arm_dex", "cameras": [ "front_camera", "robot_camera_0", "robot_camera_1" ] } }, "eval_episodes": 50, "eval_seed_base": 1000, "evaluated_steps": [ 50000 ], "selection": "final checkpoint only (no checkpoint selection)" }