如何使用 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 支持，找到了答案：

The TensorFlow estimator has as a base class sagemaker.estimator.Framework which in turn has as a base class sagemaker.estimator.EstimatorBase which accepts the parameter train_max_run which accepts a value in seconds and defaults to 86,400 or 24hs .

因此， TensorFlow估計器的初始化以傳遞最大訓練運行時間的自定義值如下所示：

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

Answer 2

讓我對@Franco 的答案進行一點更正。

sagemaker.estimator.EstimatorBase的參數名稱是max_run=86400以分配最大運行時間。