Как указать максимальное время выполнения с помощью оценщика TensorFlow с Python Sagemaker SDK?

Question

Используя Python Sagemaker SDK, можно запустить задание обучения с помощью TensorFlow со следующим кодом:

import sagemaker
from sagemaker.tensorflow import TensorFlow

sess = sagemaker.Session()
tf_estimator = TensorFlow(...)
tf_estimator.fit(...)

Можно ли указать максимальное время выполнения обучения где-нибудь в этом скрипте?

Answer 1

Нашел ответ благодаря поддержке AWS:

Оценщик TensorFlow имеет в качестве базового класса sagemaker.estimator.Framework, который, в свою очередь, имеет в качестве базового класса sagemaker.estimator.EstimatorBase, который принимает параметр train_max_run, который принимает значение в секундах и по умолчанию 86,400 или 24hs.

Таким образом, инициализация оценщика TensorFlow для передачи пользовательского значения для максимального времени выполнения обучения будет выглядеть так:

MAX_TRAINING_TIME = 99999
tf_estimator = TensorFlow(..., train_max_run=MAX_TRAINING_TIME)