Reproduce saved demo in simulator again

Open alesspalma opened this issue 6 months ago • 4 comments

Hello, thanks for your work. Is there the possibility to exactly resimulate in the simulator a given demo generated with the default rlbench/dataset_generator.py? How to do it?

Jun 26 '25 23:06 alesspalma

You can set live_demos=False, and set task_env dataset_root as well, I reproduce one successful just now

Jul 02 '25 03:07 iamstupidd

@iamstupidd can you share the code?

Jul 02 '25 07:07 alesspalma

Simple code is, task_env._dataset_root = "<your_path>" demo, = task_env.get_demos(amount=1, live_demos=False) Alternative for reproduce from the very beginning, obs_dict =task_env.reset_to_demo(demo)

Jul 02 '25 07:07 iamstupidd

@iamstupidd okay and then how to execute the loaded demo action by action?

Jul 02 '25 07:07 alesspalma