backup wip

2024-03-19 18:50:04 +00:00
parent ea17f4ce50
commit 896a11f60e
16 changed files with 169 additions and 138 deletions
--- a/lerobot/common/datasets/abstract.py
+++ b/lerobot/common/datasets/abstract.py
@@ -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",
        }