set_iteration_per_loop

函数原型

def set_iteration_per_loop(sess, train_op, iterations_per_loop=1)

功能说明

设置sess.run模式下小循环次数,即每次sess.run()在Device侧执行训练迭代的次数,可以减少Host与Device间的交互次数,缩短训练时长。

使用约束

由于该接口中有改图的操作,如果图无法修改(例如冻结了图或者使用tf.train.Supervisor创建session等),则无法使用set_iteration_per_loop接口设置大小循环。此种情况下请使用create_iteration_per_loop_varload_iteration_per_loop_var

参数说明

参数名

输入/输出

描述

sess

输入

已经创建的TensorFlow会话。

train_op

输入

更新变量或梯度的操作。

iterations_per_loop

输入

每次sess.run(),在Device侧执行训练迭代的次数,默认为1,且训练迭代总次数必须为iterations_per_loop的整数倍。

混合计算模式(mix_compile_mode为True)时,iterations_per_loop必须为1。

返回值

返回一个算子,供用户通过sess.run(op)调用。