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

用法

@contextlib.contextmanager
managed_session(
    master='', config=None, start_standard_services=True,
    close_summary_writer=True
)

返回托管會話的上下文管理器。

此上下文管理器創建並自動恢複會話。它可以選擇啟動處理檢查點和摘要的標準服務。它監視從with 塊或服務引發的異常，並根據需要停止主管。

上下文管理器通常按如下方式使用：

def train():
  sv = tf.compat.v1.train.Supervisor(...)
  with sv.managed_session(<master>) as sess:
    for step in range(..):
      if sv.should_stop():
        break
      sess.run(<my training op>)
      ...do other things needed at each training step...

當塊退出時，從with 塊或服務線程之一引發的異常再次引發。這是在停止所有線程並關閉會話之後完成的。例如，AbortedError 異常，在分布式模型中的一個工人搶占的情況下引發，當塊退出時再次引發。

如果你想在搶占的情況下重試訓練循環，你可以這樣做：

def main(...):
  while True
    try:
      train()
    except tf.errors.Aborted:
      pass

作為一種特殊情況，用於控製流的異常，例如報告輸入隊列已用盡的OutOfRangeError，不會再次從with 塊中引發：它們表示訓練循環的幹淨終止，並被視為正常終止。