ARISE-Initiative / robomimic

robomimic: A Modular Framework for Robot Learning from Demonstration
MIT License
592 stars 181 forks source link

Replay dataset #156

Closed RuiyuWANG closed 3 months ago

RuiyuWANG commented 3 months ago

Hi, I wonder how to replay the robomimic dataset on a customized environment. Have some trouble resetting the new environment with the initial state from the dataset.

amandlek commented 3 months ago

In general, it will be difficult to replay a dataset on a new customized environment, since things may have changed in the environment. You could probably manually reset the robot and object states if you want, but there won't be a great way to do this kind of thing automatically.