Fix a bug

ikostrikov · Mar 3, 2020 · 84a7582 · 84a7582
1 parent d8c758d
commit 84a7582
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/a2c_ppo_acktr/envs.py b/a2c_ppo_acktr/envs.py
@@ -230,7 +230,7 @@ def __init__(self, venv, nstack, device=None):
     def step_wait(self):
         obs, rews, news, infos = self.venv.step_wait()
         self.stacked_obs[:, :-self.shape_dim0] = \
-            self.stacked_obs[:, self.shape_dim0:]
+            self.stacked_obs[:, self.shape_dim0:].clone()
         for (i, new) in enumerate(news):
             if new:
                 self.stacked_obs[i] = 0