From 69dc8165ae5e778087d0f6d8a9c530ac71a73550 Mon Sep 17 00:00:00 2001 From: Caroline Pascal Date: Tue, 25 Nov 2025 10:49:39 +0100 Subject: [PATCH] fix(max workers) Signed-off-by: Caroline Pascal --- src/lerobot/datasets/lerobot_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/datasets/lerobot_dataset.py b/src/lerobot/datasets/lerobot_dataset.py index 6da5eb49e..8f88c3efd 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -1545,7 +1545,7 @@ class LeRobotDataset(torch.utils.data.Dataset): img_dirs.append(self._get_image_file_dir(episode_index, video_key)) fps = [self.fps]*len(video_keys) - with ProcessPoolExecutor() as executor: + with ProcessPoolExecutor(max_workers=os.process_cpu_count()-2) as executor: executor.map(encode_video_frames,img_dirs,temp_paths,fps) for img_dir in img_dirs: