restore

功能描述

mxRec训练模型加载接口。

函数原型

def restore(self, sess, reading_path)

参数说明

参数名	类型	可选/必选	说明
sess	Session	必选	需要导入模型TensorFlow的Session。
reading_path	str	必选	模型checkpoint文件的保存路径。

返回值说明

成功：None。
失败：抛出异常。

使用示例

import tensorflow as tf
saver = tf.compat.v1.train.Saver()
with tf.compat.v1.Session() as sess:
    if os.path.exists(f"./saved-model/sparse-model-{rank_id}-0"):
        saver.restore(sess, f"./saved-model/model-{rank_id}-0")

参考资源

接口调用流程及示例，参见模型训练。

父主题： 模型接口