Set starting robot pose and object positions in rollout

ARISE-Initiative / robomimic

robomimic: A Modular Framework for Robot Learning from Demonstration

MIT License

592 stars 181 forks source link

Set starting robot pose and object positions in rollout #169

Closed sarthakdas closed 2 months ago

sarthakdas commented 2 months ago

Is it possible to generate specific test cases for the robot to test the model on? in the rollout.

amandlek commented 2 months ago

You could probably use something like the reset_to function in the environment wrapper to achieve this, but in general, this will be simulation environment dependent functionality.