wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Updated implementation on MultiLeRobotDataset

This commit is contained in:
Marina Barannikov 2024-05-31 16:28:08 +00:00
parent c4870e5892
commit 212a5ab29b
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/datasets/lerobot_dataset.py
View File

@ -382,10 +382,6 @@ class MultiLeRobotDataset(torch.utils.data.Dataset):
if data_key in item: if data_key in item:
del item[data_key] del item[data_key]
if self.transform is not None:
for cam in self.camera_keys:
item[cam] = self.transform(item[cam])
return item return item
def __repr__(self): def __repr__(self):