Python tf.compat.v1.train.basic_train_loop用法及代碼示例

訓練模型的基本循環。

用法

tf.compat.v1.train.basic_train_loop(
    supervisor, train_step_fn, args=None, kwargs=None, master=''
)

supervisor tf.compat.v1.train.Supervisor 運行訓練服務。
train_step_fn 可調用以執行一個訓練步驟。反複調用 train_step_fn(session, *args **kwargs) 。
args 傳遞給 train_step_fn 的可選位置參數。
kwargs 傳遞給 train_step_fn 的可選關鍵字參數。
master 用於創建訓練課程的大師。默認為""，這會導致在本地進程中創建會話。

在循環中調用train_step_fn 來訓練模型。該函數被稱為：

train_step_fn(session, *args, **kwargs)

除了 args 和 kwargs 之外，它還傳遞了 tf.compat.v1.Session 。該函數通常在會話中運行一個訓練步驟。

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.compat.v1.train.basic_train_loop。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。