add replace=False

tanay0nSpark · Nov 3, 2020 · 338a271 · 338a271
1 parent ba997f7
commit 338a271
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ReinforcementLearning/PolicyGradient/DDPG/tensorflow2/pendulum/buffer.py b/ReinforcementLearning/PolicyGradient/DDPG/tensorflow2/pendulum/buffer.py
@@ -24,7 +24,7 @@ def store_transition(self, state, action, reward, state_, done):
     def sample_buffer(self, batch_size):
         max_mem = min(self.mem_cntr, self.mem_size)
 
-        batch = np.random.choice(max_mem, batch_size)
+        batch = np.random.choice(max_mem, batch_size, replace=False)
 
         states = self.state_memory[batch]
         states_ = self.new_state_memory[batch]