From 9cfd56587ee96df7b3eb3ea0d0fad1356d276aab Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Tue, 25 Nov 2025 12:12:37 +0100 Subject: [PATCH] fix(num processes) --- 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 e53e9e3cf..c88cfddbf 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -1546,7 +1546,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(max_workers=os.cpu_count()-2) as executor: + with ProcessPoolExecutor(max_workers=len(video_keys)) as executor: executor.map(encode_video_frames,img_dirs,temp_paths,fps) for img_dir in img_dirs: