Removed batch_encoding_Size from record.py

This commit is contained in:
Michel Aractingi
2025-07-18 17:56:42 +02:00
parent ec40fc41b5
commit 8ffc00dbcd
-2
View File
@@ -301,7 +301,6 @@ def record(cfg: RecordConfig) -> LeRobotDataset:
dataset = LeRobotDataset( dataset = LeRobotDataset(
cfg.dataset.repo_id, cfg.dataset.repo_id,
root=cfg.dataset.root, root=cfg.dataset.root,
batch_encoding_size=cfg.dataset.video_encoding_batch_size,
) )
if hasattr(robot, "cameras") and len(robot.cameras) > 0: if hasattr(robot, "cameras") and len(robot.cameras) > 0:
@@ -322,7 +321,6 @@ def record(cfg: RecordConfig) -> LeRobotDataset:
use_videos=cfg.dataset.video, use_videos=cfg.dataset.video,
image_writer_processes=cfg.dataset.num_image_writer_processes, image_writer_processes=cfg.dataset.num_image_writer_processes,
image_writer_threads=cfg.dataset.num_image_writer_threads_per_camera * len(robot.cameras), image_writer_threads=cfg.dataset.num_image_writer_threads_per_camera * len(robot.cameras),
batch_encoding_size=cfg.dataset.video_encoding_batch_size,
) )
# Load pretrained policy # Load pretrained policy