From c304474f6a550879880f4e851f9445634f4a05a3 Mon Sep 17 00:00:00 2001 From: Thomas Wolf Date: Sat, 8 Jun 2024 12:17:19 +0200 Subject: [PATCH] Update examples/real_robot_example/0_record_training_data.py Co-authored-by: Remi --- .../real_robot_example/0_record_training_data.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/real_robot_example/0_record_training_data.py b/examples/real_robot_example/0_record_training_data.py index 5412002f..8c392941 100644 --- a/examples/real_robot_example/0_record_training_data.py +++ b/examples/real_robot_example/0_record_training_data.py @@ -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." )