tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Support for converting OpenX datasets from RLDS format to LeRobotDataset (#354)

Browse Source 
Signed-off-by: youliangtan <tan_you_liang@hotmail.com>
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: youliangtan <tan_you_liang@hotmail.com>
Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Michel Aractingi
2024-08-27 09:07:00 +02:00
committed by GitHub
parent aad59e6b6b
commit eb4c505cff
12 changed files with 2329 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/scripts/save_dataset_to_safetensors.py
View File

@@ -84,5 +84,7 @@ if __name__ == "__main__":
"lerobot/pusht",
"lerobot/aloha_sim_insertion_human",
"lerobot/xarm_lift_medium",
"lerobot/nyu_franka_play_dataset",
"lerobot/cmu_stretch",
]:
save_dataset_to_safetensors("tests/data/save_dataset_to_safetensors", repo_id=dataset)