admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(num processes)

Browse Source
This commit is contained in:
CarolinePascal
2025-11-25 12:12:37 +01:00
committed by Michel Aractingi
parent ff8584a025
commit 9cfd56587e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/lerobot_dataset.py
+1 -1
View File
@@ -1546,7 +1546,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
img_dirs.append(self._get_image_file_dir(episode_index, video_key)) img_dirs.append(self._get_image_file_dir(episode_index, video_key))
fps = [self.fps]*len(video_keys) 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) executor.map(encode_video_frames,img_dirs,temp_paths,fps)
for img_dir in img_dirs: for img_dir in img_dirs: