Python tf.compat.v1.train.MonitoredSession.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。