admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-21 11:39:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

add localdir

Browse Source
This commit is contained in:
Pepijn
2025-12-02 17:26:44 +01:00
parent e1ae51b02a
commit cf36f4b873
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/dataset/slurm_port_egodex.py
+7 -1
View File
@@ -67,7 +67,7 @@ class PortEgoDexShards(PipelineStep):
def _get_state_for_single_frame(transforms_group, frame_idx): def _get_state_for_single_frame(transforms_group, frame_idx):
""" """
Construct 48D hand state representation from EgoDex paper. Construct 48D hand state representation from EgoDex.
State vector composition (per hand = 24D, total = 48D): State vector composition (per hand = 24D, total = 48D):
- Wrist 3D position (3) - Wrist 3D position (3)
@@ -421,6 +421,12 @@ def main():
default=100.0, default=100.0,
help="Percentage of dataset to process (e.g., 1.0 for 1%%). Useful for testing.", help="Percentage of dataset to process (e.g., 1.0 for 1%%). Useful for testing.",
) )
parser.add_argument(
"--local-dir",
type=Path,
default=None,
help="Local directory to save the LeRobot dataset. Defaults to data/local_datasets.",
)
args = parser.parse_args() args = parser.parse_args()
kwargs = vars(args) kwargs = vars(args)