diff --git a/baselines/a2c/runner.py b/baselines/a2c/runner.py index 8d0c6ec..0181472 100644 --- a/baselines/a2c/runner.py +++ b/baselines/a2c/runner.py @@ -37,9 +37,6 @@ class Runner(AbstractEnvRunner): obs, rewards, dones, _ = self.env.step(actions) self.states = states self.dones = dones - for n, done in enumerate(dones): - if done: - self.obs[n] = self.obs[n]*0 self.obs = obs mb_rewards.append(rewards) mb_dones.append(self.dones)