hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

backup wip

Browse Source
This commit is contained in:
Alexander Soare
2024-03-19 18:50:04 +00:00
parent ea17f4ce50
commit 896a11f60e
16 changed files with 169 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/abstract.py
View File

@@ -50,7 +50,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer):
def stats_patterns(self) -> dict:
return {
("observation", "state"): "b c -> c",
("observation", "image"): "b c h w -> c",
("observation", "image"): "b c h w -> c 1 1",
("action",): "b c -> c",
}

2
lerobot/common/datasets/aloha.py
View File

@@ -117,7 +117,7 @@ class AlohaExperienceReplay(AbstractExperienceReplay):
("action",): "b c -> c",
}
for cam in CAMERAS[self.dataset_id]:
d[("observation", "image", cam)] = "b c h w -> c"
d[("observation", "image", cam)] = "b c h w -> c 1 1"
return d
@property