From 53d3450a296ad9a4d97f36b1fb7ed07d11bfe4ad Mon Sep 17 00:00:00 2001 From: Pablo Samuel Castro Date: Tue, 5 Feb 2019 16:15:46 -0500 Subject: [PATCH] Make default number of checkpoints saved consistent between dqn_agent and checkpointer. PiperOrigin-RevId: 232544723 --- dopamine/agents/dqn/dqn_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dopamine/agents/dqn/dqn_agent.py b/dopamine/agents/dqn/dqn_agent.py index 01acd70b..5adb0fcd 100644 --- a/dopamine/agents/dqn/dqn_agent.py +++ b/dopamine/agents/dqn/dqn_agent.py @@ -95,7 +95,7 @@ def __init__(self, epsilon_decay_period=250000, tf_device='/cpu:*', use_staging=True, - max_tf_checkpoints_to_keep=3, + max_tf_checkpoints_to_keep=4, optimizer=tf.train.RMSPropOptimizer( learning_rate=0.00025, decay=0.95,