tf.estimator.TrainSpec
Configuration for the "train" part for the train_and_evaluate
call.
tf.estimator.TrainSpec(
input_fn, max_steps=None, hooks=None
)
TrainSpec
determines the input data for the training, as well as the
duration. Optional hooks run at various stages of training.
Args |
input_fn
|
A function that provides input data for training as minibatches.
See Premade Estimators
for more information. The function should construct and return one of
the following:
- A 'tf.data.Dataset' object: Outputs of
Dataset object must be a
tuple (features, labels) with same constraints as below.
- A tuple (features, labels): Where features is a
Tensor or a
dictionary of string feature name to Tensor and labels is a
Tensor or a dictionary of string label name to Tensor .
|
max_steps
|
Int. Positive number of total steps for which to train model.
If None , train forever. The training input_fn is not expected to
generate OutOfRangeError or StopIteration exceptions. See the
train_and_evaluate stop condition section for details.
|
hooks
|
Iterable of tf.train.SessionRunHook objects to run
on all workers (including chief) during training.
|
Raises |
ValueError
|
If any of the input arguments is invalid.
|
TypeError
|
If any of the arguments is not of the expected type.
|
Attributes |
input_fn
|
|
max_steps
|
|
hooks
|
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[null,null,["Last updated 2020-10-01 UTC."],[],[]]