Add logging statement for max_tf_checkpoints_to_keep in DQNAgent

PiperOrigin-RevId: 252111284
google · Jun 8, 2019 · 5ba5407 · 5ba5407
1 parent 10f8b31
commit 5ba5407
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dopamine/agents/dqn/dqn_agent.py b/dopamine/agents/dqn/dqn_agent.py
@@ -160,6 +160,8 @@ def __init__(self,
     tf.logging.info('\t tf_device: %s', tf_device)
     tf.logging.info('\t use_staging: %s', use_staging)
     tf.logging.info('\t optimizer: %s', optimizer)
+    tf.logging.info('\t max_tf_checkpoints_to_keep: %d',
+                    max_tf_checkpoints_to_keep)
 
     self.num_actions = num_actions
     self.observation_shape = tuple(observation_shape)