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

參數

• step_fn 具有單個類型參數的函數或方法StepContext.該函數可以使用參數的方法來執行計算並訪問原始會話。的返回值step_fn將從run_step_fn，除非請求停止。在這種情況下，下一個should_stop調用將返回 True。示例用法：

  ```python
  with tf.Graph().as_default():
    c = tf.compat.v1.placeholder(dtypes.float32)
    v = tf.add(c, 4.0)
    w = tf.add(c, 0.5)
    def step_fn(step_context):
      a = step_context.session.run(fetches=v, feed_dict={c:0.5})
      if a <= 4.5:
        step_context.request_stop()
        return step_context.run_with_hooks(fetches=w,
                                           feed_dict={c:0.1})
  
    with tf.MonitoredSession() as session:
      while not session.should_stop():
        a = session.run_step_fn(step_fn)
  ```
  Hooks interact with the `run_with_hooks()` call inside the
       `step_fn` as they do with a `MonitoredSession.run` call.

返回

• 返回 step_fn 的返回值。

拋出

• StopIteration 如果 step_fn 調用了 request_stop() 。它可能會被with tf.MonitoredSession() 捕獲以關閉會話。
• ValueError 如果 step_fn 沒有一個名為 step_context 的參數。當它屬於一個對象時，它還可以選擇具有self。