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

Update examples/real_robot_example/0_record_training_data.py 

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Thomas Wolf 2024-06-08 12:17:19 +02:00 committed by GitHub
parent e53a03ca53
commit c304474f6a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/real_robot_example/0_record_training_data.py
View File

@ -21,12 +21,12 @@ from lerobot.scripts.push_dataset_to_hub import push_meta_data_to_hub, push_vide
# parse the repo_id name via command line
parser = argparse.ArgumentParser()
parser.add_argument("--repo_id", type=str, default="blue_red_sort")
parser.add_argument("--num_episodes", type=int, default=2)
parser.add_argument("--num_frames", type=int, default=400)
parser.add_argument("--num_workers", type=int, default=16)
parser.add_argument("--keep_last", action="store_true")
parser.add_argument("--push_to_hub", action="store_true")
parser.add_argument("--repo-id", type=str, default="blue_red_sort")
parser.add_argument("--num-episodes", type=int, default=2)
parser.add_argument("--num-frames", type=int, default=400)
parser.add_argument("--num-workers", type=int, default=16)
parser.add_argument("--keep-last", action="store_true")
parser.add_argument("--push-to-hub", action="store_true")
parser.add_argument(
"--revision", type=str, default=CODEBASE_VERSION, help="Codebase version used to generate the dataset."
)