hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

WIP

Browse Source
This commit is contained in:
Remi Cadene
2024-12-11 09:09:14 -08:00
parent c6e9a3dc24
commit ebe0bfad77
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lerobot/scripts/manage_dataset.py
View File

@@ -53,6 +53,9 @@ if __name__ == "__main__":
help="Use local files only. By default, this script will try to fetch the dataset from the hub if it exists.", help="Use local files only. By default, this script will try to fetch the dataset from the hub if it exists.",
) )
############################################################################
# consolidate
parser_conso = subparsers.add_parser("consolidate", parents=[base_parser]) parser_conso = subparsers.add_parser("consolidate", parents=[base_parser])
parser_conso.add_argument( parser_conso.add_argument(
"--batch-size", "--batch-size",
@@ -67,6 +70,8 @@ if __name__ == "__main__":
help="Number of processes of Dataloader for computing the dataset statistics.", help="Number of processes of Dataloader for computing the dataset statistics.",
) )
############################################################################
# push_to_hub
parser_push = subparsers.add_parser("push_to_hub", parents=[base_parser]) parser_push = subparsers.add_parser("push_to_hub", parents=[base_parser])
parser_push.add_argument( parser_push.add_argument(
"--tags", "--tags",
@@ -108,3 +113,16 @@ if __name__ == "__main__":
elif mode == "push_to_hub": elif mode == "push_to_hub":
private = kwargs.pop("private") == 1 private = kwargs.pop("private") == 1
dataset.push_to_hub(private=private, **kwargs) dataset.push_to_hub(private=private, **kwargs)
elif mode == "remove_episode":
remove_episode(**kwargs)
elif mode == "delete_dataset":
delete_dataset()
elif mode == "_episode":