Fix policy defaults (#113)

2024-04-29 08:26:59 +01:00
parent 791506dfb8
commit ccffa9e406
3 changed files with 13 additions and 2 deletions
--- a/lerobot/common/policies/act/configuration_act.py
+++ b/lerobot/common/policies/act/configuration_act.py
@@ -77,7 +77,7 @@ class ActionChunkingTransformerConfig:
    # Normalization / Unnormalization
    input_normalization_modes: dict[str, str] = field(
        default_factory=lambda: {
-            "observation.image": "mean_std",
+            "observation.images.top": "mean_std",
            "observation.state": "mean_std",
        }
    )