文档章节

小白学Tensorflow之模型保存和加载

AllenOR灵感
 AllenOR灵感
发布于 2017/09/10 01:19
字数 262
阅读 8
收藏 0

今天学习了一下Tensorflow模型的保存和加载,查看了API文档,但是没有很理解,所以从网上找了一个比较简单的实现。

比如,我们需要保存的模型是参数v1v2,那么只需要使用下列的保存代码save_model.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import tensorflow as tf
v1 = tf.Variable(1.1, name="v1")
v2 = tf.Variable(1.2, name="v2")

init = tf.initialize_all_variables()
saver = tf.train.Saver()
with tf.Session() as sess:
    sess.run(init)
    print v2.eval(sess)
    save_path="model.ckpt"
    saver.save(sess,save_path)
    print "Model stored...."

如果,我们要恢复模型,并且把他们导入到变量中,那么首先定义两个参数v3v4,给他们取名叫v1v2。注意,这里必须要给v3v4取名为v1v2,因为我们保存的模型中给变量取的名字就是v1v2。那么,模型恢复的代码为restore_model.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import tensorflow as tf

v3 = tf.Variable(0.0, name="v1")
v4 = tf.Variable(0.0, name="v2")

saver = tf.train.Saver()

with tf.Session() as sess:

    save_path="model.ckpt"

    saver.restore(sess, save_path)
    print("Model restored.")
    print sess.run(v3)

本文转载自:http://www.jianshu.com/p/239c6c62f619

共有 人打赏支持
AllenOR灵感
粉丝 11
博文 2635
码字总数 83001
作品 0
程序员
私信 提问
如何部署tensorflow训练的模型

最近深度学习算法被广泛研究和应用,而tensorflow则是被应用最为广泛的工具。tensorflow训练的模型被应用在线上时,主要有3种方式(本文主要讨论java方向的应用): 1:java代码重写预测代码(...

lirainbow0
05/29
0
0
在Tensorflow Serving上部署基于LSTM的文本分类模型

一些重要的概念 Servables Servables 是客户端请求执行计算的基础对象,大小和粒度是灵活的。 Servables 不会管理自己的运行周期。 典型的Servables包括: Servable Versions Tensorflow ser...

liyonghong
02/02
0
0
TensorFlow保存和恢复模型的方法总结

使用TensorFlow训练模型的过程中,需要适时对模型进行保存,以及对保存的模型进行restore,以方便后续对模型进行处理。比如进行测试,或者部署;比如拿别的模型进行fine-tune,等等。当然,直...

Alex142857
2017/10/25
0
5
TensorFlow Lite 正式发布,谷歌移动端深度学习框架

今年 5 月,谷歌曾在 I/O 大会上宣布即将推出 TensorFlow Lite,现在,谷歌终于发布了新工具的开发者预览版本,这是一款 TensorFlow 用于移动设备和嵌入式设备的轻量级解决方案。 TensorFlow...

达尔文
2017/11/15
3.4K
5
【webAI】Tensorflow.js加载预训练的model

环境准备 win10 python3.6 pip install tensorflow pip install tensorflowjs 训练并保存tensorflow模型为saved_model...

ns2250225
04/21
0
0

没有更多内容

加载失败,请刷新页面

加载更多

day02:管道符、shell及环境变量

1、管道符:"|" 用于将前一个指令的输出作为后一个指令的输入,且管道符后面跟的是命令(针对文档的操作):cat less head tail grep cut sort wc uniq tee tr split sed awk等) [root@localho...

芬野de博客
11分钟前
4
0
Kubernetes系列——Kubernetes 组件、对象(二)

一、Kubernetes 组件 介绍了Kubernetes集群所需的各种二进制组件。 Master 组件 Master组件提供集群的管理控制中心。Master组件可以在集群中任何节点上运行。但是为了简单起见,通常在一...

吴伟祥
20分钟前
4
0
Flink-数据流编程模型

1、抽象等级 Flink提供了不同级别的抽象来开发流/批处理应用程序。 1) 低层级的抽象 最低层次的抽象仅仅提供有状态流。它通过Process函数嵌入到DataStream API中。它允许用户自由地处理来自一...

liwei2000
38分钟前
5
0
Java开发Swing实战JFrame和JTabbedPane容器的用法详细解析

概述: 项目是一个桌面程序,涉及标签和按钮组件、布局管理器组件、面板组件、列表框和下拉框组件等组件,以及Swing事件处理机制。 下面先从最基础的界面开始。 /** * @author: lishuai * @...

金铭鼎IT教育
43分钟前
14
0
flask 之旅

环境 为了正确地跑起来,你的应用需要依赖许多不同的软件。 就算是再怎么否认这一点的人,也无法否认至少需要依赖Flask本身。 你的应用的运行环境,在当你想要让它跑起来时,是至关重要的。 ...

hblt-j
43分钟前
10
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部