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


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)

除了 argskwargs 之外,它还传递了 tf.compat.v1.Session 。该函数通常在会话中运行一个训练步骤。

相关用法


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