Advertisement

模型持久化为PB文件形式

阅读量：

.ckpt文件用于存储模型的完整信息，然而在对模型进行测试时，仅需保留模型的架构及参数变量的具体数值，因为在该场景下只需通过输入层进行前向传播运算以获得输出层的结果，类似变量初始化过程，与反向传播相关的节点或用于模型保存的辅助节点则无需保留。PB文件具备语言无关性

保存PB文件的样例

复制代码 
    import tensorflow as tf
    #graph_util模块定义在tensorflow/python/framework/graph_util.py
    from tensorflow.python.framework import graph_util
    
    a = tf.Variable(tf.constant(1.0, shape=[1]), name="a")
    b = tf.Variable(tf.constant(2.0, shape=[1]), name="b")
    result = a + b
    init_op = tf.global_variables_initializer()
    
    with tf.Session() as sess:
    sess.run(init_op)
    # 导出主要记录了TensorFlow计算图上节点信息的GraphDef部分
    # 使用get

全部评论 (0)

还没有任何评论哟~

相关文章推荐