diff --git a/src/lerobot/datasets/lerobot_dataset.py b/src/lerobot/datasets/lerobot_dataset.py index 381c91b70..3afff979a 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -1454,7 +1454,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: