tensorflow中加载模型两种方法

    在使用tensorflow过程中,把模型保存下来稍后使用,是常见的需求。初步研究,各种资料混杂,读者容易不知所云,这里介绍两种方法。

本文方法在 TensorFlow 1.4 版本基础上实验。




首先定义一个简单的模型,分别采用两种方法还原模型。

train.py

这个简单的模型是 y = w + x. 其中 x 是输入, y 是输出, w采用了随机一个权重。

输出结果:

w [[-0.07159261 -0.33205539]]

y [[ 5.92840719  6.66794443]]

现在可以看到,文件夹下多了4个文件:



1. 还原变量的角度

把W还原出来,然后可以自己定义 y = w + x

输出结果 [[-0.07159261 -0.33205539]],可见,这就是 w。

然后就可以利用 w ,重构模型。

这个方法需要重新定义原模型中的 变量,加载模型后,重新定义的变量被保存的值赋值。

2. 还原图的角度

如果不想重新定义变量,可以把图还原,如下:

输出结果:

w [[-0.07159261 -0.33205539]]

[[ 4.92840719  5.66794443]]

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 本系列文章面向深度学习研发者,希望通过Image Caption Generation,一个有意思的具体任务,深入...
    imGeek阅读 1,860评论 0 8
  • 一、实验目的 学习使用 weka 中的常用分类器,完成数据分类任务。 二、实验内容 了解 weka 中 explo...
    yigoh阅读 8,648评论 5 4
  • 今天是星期二,也是军训的第十四天。今天,我们排练了明天的结营仪式,我感觉我们排练的还是不错的。排练的内容有手语操,...
    WMY王明阳阅读 144评论 0 0
  • (一) 哲学就如长途中午夜的旅馆,那一盏寂静的灯。如激动地灵魂捧起一把故乡的土壤,那热忱的泪水潸潸落地。似乎是一场...
    博古林V雙木居士阅读 787评论 1 6
  • 被描述的对象 行尸走肉 生活失去精神 夜的帷幕落下 才见光芒 当饥肠辘辘 路边的老鼠在觅食 我曾不止一次好奇 为什...
    谭剑飞阅读 225评论 0 1