Checking branch push

2023-04-15 15:54:09 +02:00 · 2023-04-15 15:54:09 +02:00 · 9a2e9f420b
commit 9a2e9f420b
parent 9aa07fed6a
1 changed files with 1 additions and 0 deletions
--- a/DPI/utils.py
+++ b/DPI/utils.py
@ -200,6 +200,7 @@ class ReplayBuffer:
        variable = getattr(buffer, variable)
        non_zero_indices = np.nonzero(buffer.episode_count)[0]
        variable = variable[non_zero_indices]
+        
        if obs:
            variable = variable.reshape(-1, self.args.episode_length,
                                        self.args.frame_stack*self.args.channels,