mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 11:40:18 +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:
@@ -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