Python MujocoState.from_flattened示例

示例#1

文件： lookback.py 项目： wingsweihua/adversarial-policies-sync

    def set_state(self, x, forward=True):
        """Restores qpos and qvel, calling forward() to derive other values.

        :param forward (bool) whether to call forward on the environment.
        """
        state = MujocoState.from_flattened(x, self.sim)
        state.set_mjdata(self.sim.data)
        if forward:
            self.sim.model.forward()  # put mjData in consistent state

示例#2

 def set_state(self, x):
     state = MujocoState.from_flattened(x, self.sim)
     state.set_mjdata(self.sim.data)
     self.sim.forward()  # put mjData in consistent state

示例#3

文件： monte_carlo.py 项目： pranshulnarang/adversarial-policies

 def set_state(self, x):
     """Restores qpos and qvel, calling forward() to derive other values."""
     state = MujocoState.from_flattened(x, self.sim)
     state.set_mjdata(self.sim.data)
     self.sim.forward()  # put mjData in consistent state