update

bob7783 · bob7783 · commit f108f1caaeb3 · 2019-02-08T22:52:43.000-05:00
diff --git a/rl2/atari/dqn_tf.py b/rl2/atari/dqn_tf.py
@@ -60,7 +60,7 @@ def update_state(state, obs_small):
 
 
 class ReplayMemory:
-  def __init__(self, size=500000, frame_height=IM_SIZE, frame_width=IM_SIZE, 
+  def __init__(self, size=MAX_EXPERIENCES, frame_height=IM_SIZE, frame_width=IM_SIZE, 
                agent_history_length=4, batch_size=32):
     """
     Args:
diff --git a/rl2/atari/dqn_theano.py b/rl2/atari/dqn_theano.py
@@ -59,7 +59,7 @@ def update_state(state, obs):
 
 
 class ReplayMemory:
-  def __init__(self, size=500000, frame_height=IM_SIZE, frame_width=IM_SIZE, 
+  def __init__(self, size=MAX_EXPERIENCES, frame_height=IM_SIZE, frame_width=IM_SIZE, 
                agent_history_length=4, batch_size=32):
     """
     Args:
