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 parent
a112ee69f6
commit
b94239234a
2 files changed
+3
-1
No files matched your search
@@ -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