get_initializer

功能描述

获取初始化iterator的TensorFlow算子（tf.Operation），该算子需要通过sess.run()使用。

函数原型

def get_initializer(is_training)

参数说明

返回值说明

• 成功：返回初始化iterator的TensorFlow算子（tf.Operation）。
• 失败：抛出异常。

使用示例

import tensorflow as tf
from mx_rec.util.initialize import get_initializer, set_initializer

# 创建对应的iterator，比如train模式下，创建train_iterator；predict模式下创建eval_iterator，具体创建方式参考Tensorflow
set_initializer(True, train_iterator.initializer)  # train模式下就为train_iterator；predict模式下为eval_iterator
with tf.compat.v1.Session() as sess:
    sess.run(tf.compat.v1.global_variables_initializer())
    sess.run(get_initializer(True))

参考资源

接口调用流程及示例，参见自动改图。