tf.compat.v1.train.basic_train_loop
Простой цикл для обучения модели.
tf.compat.v1.train.basic_train_loop(
supervisor, train_step_fn, args=None, kwargs=None, master=''
)
Вызывает train_step_fn в цикле для обучения модели. Функция вызывается следующим образом:
train_step_fn(session, *args, **kwargs)
Ей передается tf.compat.v1.Session, а также args и kwargs. Обычно функция выполняет один шаг обучения в сеансе.
| Аргументы | |
|---|---|
supervisor | tf.compat.v1.train.Supervisor для запуска служб обучения. |
train_step_fn | Функция обратного вызова для выполнения одного шага обучения. Вызывается многократно как train_step_fn(session, *args **kwargs). |
args | Необязательные позиционные аргументы, передаваемые в train_step_fn. |
kwargs | Необязательные именованные аргументы, передаваемые в train_step_fn. |
master | Главный узел, используемый для создания сеанса обучения. По умолчанию — "", что приводит к созданию сеанса в локальном процессе. |
© 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/api_docs/python/tf/compat/v1/train/basic_train_loop