Skip to content

Gr00t: training ends in error when images are not the same size #2334

@gyger

Description

@gyger

I tried refining Gr00t using the scripts provided in lerobot. Unfortunately my recorded data was with different input image resolutions. Not sure if you think it makes sense to support this, but would be great.

It then cancels in processor_groot.py, when around line 300 all the camera images are stacked.

I guess: aspect respecting rescaling, or just rescaling or a better way (perhaps it already exists) to control the input pipeline for video in training and inference would be great. I am probably a little bit ignorant here, did not write my own code just used the scripts provided by lerobot.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn’t working correctlypoliciesItems related to robot policies

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions