vimalabs / VIMA

Official Algorithm Implementation of ICML'23 Paper "VIMA: General Robot Manipulation with Multimodal Prompts"
MIT License
778 stars 87 forks source link

Question: how to specific state or restore the state of the environment? #40

Closed LongXinKou closed 10 months ago

LongXinKou commented 1 year ago

Hi

setting ramdom seeds doesn't semms to work `seed = 42 np.random.seed(seed) env_1 = TimeLimitWrapper( ResetFaultToleranceWrapper( make( cfg.task, modalities=["segm", "rgb"], task_kwargs=PARTITION_TO_SPECS["test"][cfg.partition][cfg.task], seed=seed,

render_prompt=True,

            display_debug_window=True,
            hide_arm_rgb=True,
            # record_gui=True,
        )
    ),
    bonus_steps=2,
)`

..... can i use env.state to save current env state, and load it next time?