Skip to content

Commit 9acca5d

Browse files
authored
Merge pull request rushter#52 from ndrwmlnk/master
bug fix self.memory_limit
2 parents c0eacae + 8f49f66 commit 9acca5d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mla/rl/dqn.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def train(self, render=False):
118118
break
119119

120120
# Remove old entries from replay memory
121-
if len(self.replay) > self.memory_limit:
121+
while len(self.replay) > self.memory_limit:
122122
self.replay.pop(0)
123123

124124
self.epsilon = self.min_epsilon + (1.0 - self.min_epsilon) * np.exp(-self.decay * ep)

0 commit comments

Comments
 (0)