wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update push_to_hub

This commit is contained in:
Thomas Wolf 2024-05-31 11:20:25 +02:00
parent 0f50e033e9
commit 11cde3bcd4
2 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -231,7 +231,8 @@ huggingface-cli login --token ${HUGGINGFACE_TOKEN} --add-to-git-credential
Then move your dataset folder in `data` directory (e.g. `data/aloha_static_pingpong_test`), and push your dataset to the hub with:
```bash
python lerobot/scripts/push_dataset_to_hub.py \
--data-dir data \
--raw-dir data_raw \
--out-dir out_data \
--dataset-id aloha_static_pingpong_test \
--raw-format aloha_hdf5 \
--community-id lerobot

20
lerobot/scripts/push_dataset_to_hub.py
View File

@ -21,7 +21,8 @@ installation of neural net specific packages like pytorch, tensorflow, jax.
Example:
```
python lerobot/scripts/push_dataset_to_hub.py \
--data-dir data \
--raw-data data_raw \
--out-dir data \
--dataset-id pusht \
--raw-format pusht_zarr \
--community-id lerobot \
@ -31,7 +32,8 @@ python lerobot/scripts/push_dataset_to_hub.py \
--debug 1
python lerobot/scripts/push_dataset_to_hub.py \
--data-dir data \
--raw-dir data_raw \
--out-dir data \
--dataset-id xarm_lift_medium \
--raw-format xarm_pkl \
--community-id lerobot \
@ -41,7 +43,8 @@ python lerobot/scripts/push_dataset_to_hub.py \
--debug 1
python lerobot/scripts/push_dataset_to_hub.py \
--data-dir data \
--raw-dir data_raw \
--out-dir data \
--dataset-id aloha_sim_insertion_scripted \
--raw-format aloha_hdf5 \
--community-id lerobot \
@ -51,7 +54,8 @@ python lerobot/scripts/push_dataset_to_hub.py \
--debug 1
python lerobot/scripts/push_dataset_to_hub.py \
--data-dir data \
--raw-dir data_raw \
--out-dir data \
--dataset-id umi_cup_in_the_wild \
--raw-format umi_zarr \
--community-id lerobot \
@ -287,10 +291,10 @@ def main():
"--save-to-disk",
type=int,
default=1,
help="Save the dataset in the directory specified by `--data-dir`.",
help="Save the dataset in the directory specified by `--out-dir`.",
)
parser.add_argument(
"--tests-data-dir",
"--tests-out-dir",
type=Path,
default="tests/data",
help="Directory containing tests artifacts datasets.",
@ -298,8 +302,8 @@ def main():
parser.add_argument(
"--save-tests-to-disk",
type=int,
default=1,
help="Save the dataset with 1 episode used for unit tests in the directory specified by `--tests-data-dir`.",
default=0,
help="Save the dataset with 1 episode used for unit tests in the directory specified by `--tests-out-dir`.",
)
parser.add_argument(
"--fps",