文档章节

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

AllenOR灵感
 AllenOR灵感
发布于 2017/09/10 01:19
字数 262
阅读 7
收藏 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灵感
粉丝 10
博文 2634
码字总数 82983
作品 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
【webAI】Tensorflow.js加载预训练的model

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

ns2250225
04/21
0
0
如何使用 TensorFlow mobile 将 PyTorch 和 Keras 部署到移动设备

雷锋网(公众号:雷锋网)按:本文为雷锋字幕组编译的技术博客,原标题 Deploying PyTorch and Keras Models to Android with TensorFlow Mobile ,作者为 John Olafenwa 。 翻译 | 于志鹏 整理...

雷锋字幕组
07/12
0
0
TensorFlow保存和恢复模型的方法总结

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

Alex142857
2017/10/25
0
5

没有更多内容

加载失败,请刷新页面

加载更多

70.shell的函数 数组 告警系统需求分析

20.16/20.17 shell中的函数 20.18 shell中的数组 20.19 告警系统需求分析 20.16/20.17 shell中的函数: ~1. 函数就是把一段代码整理到了一个小单元中,并给这个小单元起一个名字,当用到这段...

王鑫linux
今天
2
0
分布式框架spring-session实现session一致性使用问题

前言:项目中使用到spring-session来缓存用户信息,保证服务之间session一致性,但是获取session信息为什么不能再服务层获取? 一、spring-session实现session一致性方式 用户每一次请求都会...

WALK_MAN
今天
6
0
C++ yield()与sleep_for()

C++11 标准库提供了yield()和sleep_for()两个方法。 (1)std::this_thread::yield(): 线程调用该方法时,主动让出CPU,并且不参与CPU的本次调度,从而让其他线程有机会运行。在后续的调度周...

yepanl
今天
4
0
Java并发编程实战(chapter_3)(线程池ThreadPoolExecutor源码分析)

这个系列一直没再写,很多原因,中间经历了换工作,熟悉项目,熟悉新团队等等一系列的事情。并发课题对于Java来说是一个又重要又难的一大块,除非气定神闲、精力满满,否则我本身是不敢随便写...

心中的理想乡
今天
43
0
shell学习之获取用户的输入命令read

在运行脚本的时候,命令行参数是可以传入参数,还有就是在脚本运行过程中需要用户输入参数,比如你想要在脚本运行时问个问题,并等待运行脚本的人来回答。bash shell为此提 供了read命令。 ...

woshixin
今天
4
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部