当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


用法

run_step_fn(
    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

使用步进函数运行操作。

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.train.SingularMonitoredSession.run_step_fn。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。