参数存取 tensorflow学习

tensorflow学习(5)参数存取

Posted by 高庆东 on September 6, 2018

参数存取

需要注意的的是保存的是所有变量取的也是所有变量 取的时候一定要按存的顺去来取

这个涉及到fintune问题

import tensorflow as tf
import numpy as np
# 保存参数
w=tf.Variable([[1,2,3],[4,5,6]],dtype=tf.float32,name="weight")
b=tf.Variable([[1,2,3]],dtype=tf.float32,name='bias')

init=tf.global_variables_initializer()
seen=tf.Session()

save=tf.train.Saver()
seen.run(init)

save_path=save.save(seen,"Download/save.ckpt")

#提取参数
W = tf.Variable(np.arange(6).reshape((2, 3)), dtype=tf.float32, name="weights")
b = tf.Variable(np.arange(3).reshape((1, 3)), dtype=tf.float32, name="biases")

saver = tf.train.Saver()
with tf.Session() as sess:
    # 提取变量
    saver.restore(sess, "my_net/save_net.ckpt")
    print("weights:", sess.run(W))
    print("biases:", sess.run(b))
CATALOG
    FEATURED TAGS
    c语言 c++ 面向对象 指针 容器 python 函数 数据结构 回归 损失函数 神经网络 机器学习 似然函数 极大似然 标准化 深度学习 卷积网络 参数估计 beta分布 数据处理 gradio 模型工程化 网页 模型加速工具 c++实现 变量 占位符 tensorflow 线性回归 学习tensorflow HMM RNN 强化学习 LSTM pandas 不定长序列损失 pytorch 目标检测 RPN 非极大值抑制 ROIpooling VGG16 Transformer BERT Python 装饰器 方法 Pytorch FPN 图像分类 CNN 多模态 生成 GPT Tranformer 生成模型 audio