mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 03:30:19 +01:00
Removed TF warning when training in a distributed setting (#133)
* 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:
1
.gitignore
vendored
1
.gitignore
vendored
@@ -17,6 +17,7 @@ docs/site
|
||||
coach_env
|
||||
build
|
||||
rl_coach.egg*
|
||||
rl_coach_slim.egg*
|
||||
contrib
|
||||
test_log_*
|
||||
dist
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user