Spec-Zone.ru › TensorFlow 2.9

tf.estimator.TrainSpec

Просмотреть исходный код на GitHub

Настройка для части "обучение" вызова 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 должны быть кортежем (features, labels) с теми же ограничениями, что и ниже.
  • Кортеж (features, labels): где features является Tensor или словарем из имени строки признака к Tensor, а labels является 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 Псевдоним namedtuple для поля номер 0
max_steps Псевдоним namedtuple для поля номер 1
hooks Псевдоним namedtuple для поля номер 2
saving_listeners Псевдоним namedtuple для поля номер 3

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API