gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01:00
Code Issues Projects Releases Wiki Activity

Removed TF warning when training in a distributed setting (#133)

Browse Source 
* removed TF warning when training in a distributed setting and changed package version

* revert version back to 0.11.0
This commit is contained in:
shadiendrawis
2018-11-21 16:09:04 +02:00
committed by GitHub
parent a112ee69f6
commit b94239234a
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -17,6 +17,7 @@ docs/site
coach_env coach_env
build build
rl_coach.egg* rl_coach.egg*
rl_coach_slim.egg*
contrib contrib
test_log_* test_log_*
dist dist

3
rl_coach/architectures/tensorflow_components/distributed_tf_utils.py
View File

@@ -95,7 +95,8 @@ def create_monitored_session(target: tf.train.Server, task_index: int,
hooks=[], hooks=[],
checkpoint_dir=checkpoint_dir, checkpoint_dir=checkpoint_dir,
save_checkpoint_secs=checkpoint_save_secs, 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 return sess