fix(max workers)

Signed-off-by: Caroline Pascal <caroline8.pascal@gmail.com>
This commit is contained in:
Caroline Pascal
2025-11-25 10:49:39 +01:00
committed by Michel Aractingi
parent 021bca2ad9
commit 69dc8165ae
+1 -1
View File
@@ -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: