Reinforcement Learning, Gaming, Deep Learning.
kwargs.get("return_info", False) if return_info: obs, info = self.env.reset(**kwargs) else: obs = self.env.reset(**kwargs) obs = self.normalize(obs) if not return_info: return obs else: return obs, info def normalize(self, obs): self.obs_rms.update(obs) return (obs - self.obs_rms.mean) /