Ｔensorflow使用笔记（１）　怎么保存和使用训练好的参数和模型

最新推荐文章于 2024-08-08 18:07:23 发布
最新推荐文章于 2024-08-08 18:07:23 发布
阅读量1.3w 收藏 17
点赞数 6
分类专栏： tensorflow 文章标签： tensorflow cnn
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/yat_chiu/article/details/72851316
版权
本文介绍了在Tensorflow中如何保存和加载训练好的模型参数，重点解决了在保存过程中遇到的问题，强调了正确指定计算图和会话的重要性。同时，提供了在新脚本中加载模型并进行测试的两种方法。
摘要由CSDN通过智能技术生成

如何保存和使用训练好的模型参数

引言

最近在学习Tensorflow 构建CNN，训练需要耗费时间，把训练好的各个参数保存下来是最简便的，网上有很多教程，但是跟着教程走不一定一帆风顺还是踩了一些坑，然后自己填了一下坑

如何保存好训练的结果：

假设会话为sess，计算图为graph
网上看了很多资料，使用 

saver=tf.train.Saver()  # 不传入参数代表默认存入全部参数
file_name = 'saved_model/model.ckpt'  # 将保存到当前目录下的的saved_model文件夹下model.ckpt文件
saver.saver(sess,file_name )  # 保存好的模型文件

这样来保存模型，这样就行了吗？还不行
但是按照这个方式，开始IDE总是会报错：No Variable to save

然后我的想法是：可能要把sess传给saver是吗？下面是我的代码，和一次尝试性的修改

最低0.47元/天 解锁文章
YatChiu
关注
  • 6
    点赞
  • 踩
  • 17
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
TensorFlow 训练好模型参数的保存和恢复代码
TensorFlowNews
07-16 2206
TensorFlow 训练好模型参数的保存和恢复代码，之前就在想模型不应该每次要个结果都要重新训练一遍吧，应该训练一次就可以一直使用吧。 TensorFlow 提供了 Saver 类，可以进行保存和恢复。下面是 TensorFlow-Examples 项目中提供的保存和恢复代码。 ''' Save and Restore a model using TensorFlow.
【Tensorflow】调用已经保存好的模型继续训练(最简版)
Vincent Lai的博客
07-19 2785
import tensorflow as tf from pathlib import Path def save(path='./model/transfer_learn'): saver = tf.train.Saver() saver.save(sess, path, write_meta_graph=False) def restore(path='./model/tr...
存储Tensorflow训练网络的参数
leadai的博客
03-13 1286
正文共3565个字，预计阅读时间10分钟。上海站 | 高性能计算之GPU CUDA培训4月13-15日三天密集式训练 带你快速晋级阅读全文>训练一个神经网络的目的是啥？不就是有朝一日让它有用武之地吗？可是，在别处使用训练好的网络，得先把网络的参数（就是那些variables）保存下来，怎么保存呢？其实，tensorflow已经给我们提供了很方便的API，来帮助我们实现训练参数的存储与读取，如果想了
【机器学习】ImageNet的基本概念以及如何使用ImageNet数据集
最新发布
ya的博客
08-08 1875
ImageNet是一个大规模的视觉数据库，旨在为研究人员和开发者提供一个丰富的资源，以促进计算机视觉领域的发展。该项目始于2007年，由斯坦福大学的教授李飞飞领导，并于2009年在计算机视觉与模式识别会议（CVPR）上发布。ImageNet根据WordNet的层级结构（目前仅限于名词）组织，其中每个层级节点都由成百上千张图像来描绘
tensorflow 中的保存训练的参数以及载入参数检测test集
weixin_34310369的博客
03-09 511
如何保存训练好的参数 以前面练习的一个小例子，来保存训练好的参数： 请参考juejin.im/post/5a9fcf… import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data from tensorflow import float32 #载入数据，会自动通过一个脚本下载好数据集 mni...
TensorFlow学习笔记(2)——保存和加载训练模型参数
zchang81的博客
04-07 1395
在 Tensorflow 的训练过程中，参数通过使用 Saver 类来保存和加载模型。 Saver类可以处理图中元数据和变量数据的保存和加载。我们只需要告诉 Saver 类我们需要保存哪个图？ 哪些变量？
Tensorflow对训练模型的保存以及使用
bycare的博客
04-22 1136
（1）对训练模型的保存，将模型命名为my_net.ckp,并保存在net/文件夹下         saver = tf.train.Saver()         with tf.Session() as sess:             #保存模型             saver.save(sess,'net/my_net.ckpt')（2）对本地模型进行调用        saver ...
深度学习tensorflow实战笔记（3）VGG-16训练自己的数据并测试和保存模型
gybheroin的博客
04-03 1万+
    前面的博客介绍了如何把图像数据转换成tfrecords格式并读取，本篇博客介绍如何用自己的tfrecords格式的数据训练CNN模型，采用的模型是VGG-16。现有的教程都是在mnist或者cifar-10数据集上做的训练，如何用自己的数据集进行训练相关的资料比较少，即使有相关的也是不完整，所以做一个完成流程的学习笔记，以供以后查阅使用。首先要做的是把自己的数据转换成tfrecords文件...
TensorFlow学习笔记1
08-08
在TensorFlow中，学习和理解各种操作是至关重要的。本篇笔记主要涵盖了...以上就是TensorFlow学习笔记1中涉及的关键概念和使用技巧，掌握这些知识点将有助于深入理解和应用TensorFlow进行深度学习模型的构建和训练。
【Tensorflow教程笔记】Tensorflow模型建立与训练
nanke_4869的博客
02-20 6452
Tensorflow教程笔记 TensorFlow 基础 TensorFlow 模型建立与训练 目录Tensorflow教程笔记模型（Model）与层（Layer）Keras 的全连接层：线性变换 + 激活函数为什么模型类是重载 `call()` 方法而不是 `__call__()` 方法？ 本章介绍如何使用 TensorFlow 快速搭建动态模型。 模型的构建： tf.keras.Model 和 tf.keras.layers 模型的损失函数： tf.keras.losses 模型的优化器： tf.k
炼数成金Tensorflow课程笔记1
08-08
在"炼数成金Tensorflow课程笔记1"中，主要涵盖了TensorFlow的基本概念、代码示例，以及简单的线性和非线性回归模型。 1. **基本概念** TensorFlow 使用图（graphs）来表示计算任务，这些图由操作（operations）...
tf2使用cnn模型训练保存自己的数据（分类）
sdhdsf132452的博客
07-19 723
#!/usr/bin/env python # -*- coding: utf-8 -*- # @Time : 2021/7/19 8:22 # @Author : wutiande import numpy as np import matplotlib.image as mping import matplotlib.pyplot as plt import tensorflow as tf tf.compat.v1.set_random_seed(2021) from tensorflow.ker.
Tensorflow如何直接使用预训练模型(vgg16为例)
weixin_44633882的博客
04-06 1万+
Tensorflow如何直接使用预训练模型(vgg16为例) 本文链接： 主流的CNN模型基本都会使用VGG16或者ResNet等网络作为预训练模型，正好有个朋友和我说发给他一个VGG16的预训练模型和代码，我就整理了一下。在这里也分享一下，方便大家直接使用。 系统环境 Tensorflow-gpu 1.12.0 Python 3.5.2 资料来源 官方slim说明 https://gith...
学习TensorFlow，保存学习到的网络结构参数并调用
热门推荐
何雷
06-17 2万+
在深度学习中，不管使用那种学习框架，我们会遇到一个很重要的问题，那就是在训练完之后，如何存储学习到的深度网络的参数？在测试时，如何调用这些网络参数？针对这两个问题，本篇博文主要探索TensorFlow如何解决他们？本篇博文分为三个部分，第一是讲解tensorflow相关的函数，第二是代码例程，第三是运行结果。
tensorflow: 花卉分类
qq_25987491的博客
07-02 1万+
 本文主要通过CNN进行花卉的分类，训练结束保存模型，最后通过调用模型，输入花卉的图片通过模型来进行类别的预测。       测试平台：win 10+tensorflow 1.2       数据集：http://download.tensorflow.org/example_images/flower_photos.tgz       数据集中总共有五种花，分别放在五个文件夹下。       一...
Tensorflow-saver模型参数保存及载入
愿你远行千里，归来仍是少年
05-30 1574
深度学习框架Tensorflow学习与应用实战 1.训练模型参数(权值矩阵、偏置值)并保存到指定文件夹下 #保存训练好的模型参数(权重矩阵、偏置值等)到指定文件夹中 import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #载入数据集,one_hot:将标签转换为只有一位为1，其它...
Tensorflow 模型参数保存方法
LaiCheng的博客
06-21 4122
Tensorflow 模型参数保存方法 Tensorflow 保存模型参数分为两种，第一种是未剪枝参数，第二种是剪枝后的参数。对于未剪枝的参数，可以直接保存在TXT,剪枝后的参数可以按照稀疏矩阵保存方法进行保存。 一、未剪枝参数保存 读取参数之前需要保存模型和对应的参数，保存方法按照如下： def save_model(self): saver = tf.trai...
tensorflow如何保存best_model
Meteor的笔记
12-28 2770
背景： 在最近的实验中，用到了tensorflow做为我的模型框架。对于我而言，早已听说tensorflow的“赫赫威名”，但实际上我还属于tensorflow的一个新手，因此在实际应用中也遇到了许多问题。今天总结一下，在使用tensorflow框架时，我是如何解决保存最佳model这个问题。 问题描述： 对于保存最佳模型，我有两种思路： 在每个epoch过后，计算改epoch所得到的模型在test数据上的结果。 优点：不需要将所有epoch的model都保存下...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值