Reinforcement Learning, Gaming, Deep Learning.
self.gamma = gamma self.epsilon = epsilon def step(self, action): obs, rews, dones, infos = self.env.step(action) dones = dones.bool() self.returns = self.returns * self.gamma + rews rews = self.normalize(rews) self.returns[dones] = 0.0 return obs, rews, dones, infos def normalize(self, rews):