tf.estimator.TrainSpec

Class TrainSpec

Configuration for the "train" part for the train_and_evaluate call.

Aliases:

  • Class tf.compat.v1.estimator.TrainSpec
  • Class tf.compat.v2.estimator.TrainSpec
  • Class tf.estimator.TrainSpec

Defined in python/estimator/training.py.

TrainSpec determines the input data for the training, as well as the duration. Optional hooks run at various stages of training.

Properties

input_fn

max_steps

hooks