diff --git a/.gitignore b/.gitignore index 2f46d8b..50bcf54 100644 --- a/.gitignore +++ b/.gitignore @@ -17,6 +17,7 @@ docs/site coach_env build rl_coach.egg* +rl_coach_slim.egg* contrib test_log_* dist diff --git a/rl_coach/architectures/tensorflow_components/distributed_tf_utils.py b/rl_coach/architectures/tensorflow_components/distributed_tf_utils.py index 2ea216d..bbbbc0f 100644 --- a/rl_coach/architectures/tensorflow_components/distributed_tf_utils.py +++ b/rl_coach/architectures/tensorflow_components/distributed_tf_utils.py @@ -95,7 +95,8 @@ def create_monitored_session(target: tf.train.Server, task_index: int, hooks=[], checkpoint_dir=checkpoint_dir, save_checkpoint_secs=checkpoint_save_secs, - config=config + config=config, + log_step_count_steps=0 # disable logging of steps to avoid TF warning during inference ) return sess