Skip to content

Default ACT config causes model init failure #102

@qgallouedec

Description

@qgallouedec
from lerobot.common.policies.act.configuration_act import ActionChunkingTransformerConfig
from lerobot.common.policies.act.modeling_act import ActionChunkingTransformerPolicy

configuration = ActionChunkingTransformerConfig()
model = ActionChunkingTransformerPolicy(configuration)
Traceback (most recent call last):
  File "/Users/quentingallouedec/lerobot/p.py", line 6, in <module>
    model = ActionChunkingTransformerPolicy(configuration)
  File "/Users/quentingallouedec/lerobot/lerobot/common/policies/act/modeling_act.py", line 75, in __init__
    self.normalize_inputs = Normalize(cfg.input_shapes, cfg.normalize_input_modes, dataset_stats)
  File "/Users/quentingallouedec/lerobot/lerobot/common/policies/normalize.py", line 101, in __init__
    stats_buffers = create_stats_buffers(shapes, modes, stats)
  File "/Users/quentingallouedec/lerobot/lerobot/common/policies/normalize.py", line 31, in create_stats_buffers
    shape = tuple(shapes[key])
KeyError: 'observation.image'

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn’t working correctly

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions