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

Fix record

This commit is contained in:
Remi Cadene 2024-10-24 15:27:33 +02:00
parent 102e1d4436
commit abbf9c5bee
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/control_robot.py
View File

@ -207,6 +207,8 @@ def record(
num_image_writer_threads_per_camera: int = 4, num_image_writer_threads_per_camera: int = 4,
display_cameras: bool = True, display_cameras: bool = True,
play_sounds: bool = True, play_sounds: bool = True,
tags: str = None,
force_override: bool = False,
) -> LeRobotDataset: ) -> LeRobotDataset:
# TODO(rcadene): Add option to record logs # TODO(rcadene): Add option to record logs
listener = None listener = None
@ -237,8 +239,8 @@ def record(
dataset = LeRobotDataset.create( dataset = LeRobotDataset.create(
repo_id, repo_id,
fps, fps,
robot,
root=root, root=root,
robot=robot,
image_writer_processes=num_image_writer_processes, image_writer_processes=num_image_writer_processes,
image_writer_threads_per_camera=num_image_writer_threads_per_camera, image_writer_threads_per_camera=num_image_writer_threads_per_camera,
use_videos=video, use_videos=video,