get_initializer

功能描述

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

函数原型

def get_initializer(is_training)

参数说明

返回值说明

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

使用示例

from mx_rec.util.initialize import get_initializer, set_initializer

# 获取或者定义iterator,具体可以参考little demo，以下仅作为步骤示例，不用做实际运行
iterator = dataset.make_initializable_iterator()  # dataset需用户自己定义
set_initializer(True, iterator.initializer)
with tf.compat.v1.Session() as sess:
    sess.run(get_initializer(True))

参考资源

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