View source on GitHub |
Creates a tf.compat.v1.Session for a worker.
Inherits From: SessionCreator
tf.compat.v1.train.WorkerSessionCreator(
scaffold=None, master='', config=None, max_wait_secs=(30 * 60)
)
Args | |
---|---|
scaffold
|
A Scaffold used for gathering or building supportive ops. If
not specified a default one is created. It's used to finalize the graph.
|
master
|
String representation of the TensorFlow master to use.
|
config
|
ConfigProto proto used to configure the session.
|
max_wait_secs
|
Maximum time to wait for the session to become available. |
Methods
create_session
create_session()