tf.estimator.TrainSpec

Настройка для части "обучение" для вызова train_and_evaluate.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.estimator.TrainSpec

tf.estimator.TrainSpec(
    input_fn, max_steps=None, hooks=None, saving_listeners=None
)

TrainSpec определяет входные данные для обучения, а также продолжительность. Необязательные хуки запускаются на разных этапах обучения.

Использование:

train_spec = tf.estimator.TrainSpec(
   input_fn=lambda: 1,
   max_steps=100,
   hooks=[_StopAtSecsHook(stop_after_secs=10)],
   saving_listeners=[_NewCheckpointListenerForEvaluate(None, 20, None)])
train_spec.saving_listeners[0]._eval_throttle_secs
20
train_spec.hooks[0]._stop_after_secs
10
train_spec.max_steps
100

Аргументы
`input_fn`	Функция, которая предоставляет входные данные для обучения в виде мини-пакетов. См. Предопределенные оценщики для получения дополнительной информации. Функция должна создавать и возвращать один из следующих объектов: Объект 'tf.data.Dataset': Выходы объекта `Dataset` должны быть кортежем (признаки, метки) с теми же ограничениями, что и ниже. Кортеж (признаки, метки): где признаки — это `Tensor` или словарь имени признака типа строка к `Tensor`, а метки — это `Tensor` или словарь имени метки типа строка к `Tensor`.
`max_steps`	Целое число. Положительное число общих шагов для обучения модели. Если `None`, обучение будет продолжаться бесконечно. Обучение `input_fn` не должно генерировать `OutOfRangeError` или `StopIteration` исключений. Подробности см. в разделе условия остановки `train_and_evaluate`.
`hooks`	Итерируемый набор объектов `tf.train.SessionRunHook` для выполнения на всех рабочих узлах (включая главный) во время обучения.
`saving_listeners`	Итерируемый набор объектов `tf.estimator.CheckpointSaverListener` для выполнения на главном узле во время обучения.

Исключения
`ValueError`	Если любой из входных аргументов некорректен.
`TypeError`	Если любой из аргументов имеет несоответствующий тип.

Атрибуты
`input_fn`
`max_steps`
`hooks`
`saving_listeners`

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/estimator/TrainSpec