View source on GitHub
|
Basic loop to train a model.
tf.train.basic_train_loop(
supervisor, train_step_fn, args=None, kwargs=None, master=''
)
Calls train_step_fn in a loop to train a model. The function is called as:
train_step_fn(session, *args, **kwargs)
It is passed a tf.compat.v1.Session in addition to args and kwargs. The
function
typically runs one training step in the session.
Args | |
|---|---|
supervisor
|
tf.compat.v1.train.Supervisor to run the training services.
|
train_step_fn
|
Callable to execute one training step. Called repeatedly as
train_step_fn(session, *args **kwargs).
|
args
|
Optional positional arguments passed to train_step_fn.
|
kwargs
|
Optional keyword arguments passed to train_step_fn.
|
master
|
Master to use to create the training session. Defaults to ""
which causes the session to be created in the local process.
|