mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-04 00:27:15 +00:00
fix(os version)
Signed-off-by: Caroline Pascal <caroline8.pascal@gmail.com>
This commit is contained in:
committed by
Michel Aractingi
parent
6bc1e5186a
commit
ff8584a025
@@ -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.process_cpu_count()-2) as executor:
|
||||
with ProcessPoolExecutor(max_workers=os.cpu_count()-2) as executor:
|
||||
executor.map(encode_video_frames,img_dirs,temp_paths,fps)
|
||||
|
||||
for img_dir in img_dirs:
|
||||
|
||||
Reference in New Issue
Block a user