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

最新推荐文章于 2024-08-25 18:49:06 发布
最新推荐文章于 2024-08-25 18:49:06 发布
阅读量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
关注
专栏目录
TensorFlow 训练好模型参数的保存和恢复代码
TensorFlowNews
07-16 2209
TensorFlow 训练好模型参数的保存和恢复代码，之前就在想模型不应该每次要个结果都要重新训练一遍吧，应该训练一次就可以一直使用吧。 TensorFlow 提供了 Saver 类，可以进行保存和恢复。下面是 TensorFlow-Examples 项目中提供的保存和恢复代码。 ''' Save and Restore a model using TensorFlow.
【Tensorflow】调用已经保存好的模型继续训练(最简版)
Vincent Lai的博客
07-19 2788
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 1289
正文共3565个字，预计阅读时间10分钟。上海站 | 高性能计算之GPU CUDA培训4月13-15日三天密集式训练 带你快速晋级阅读全文>训练一个神经网络的目的是啥？不就是有朝一日让它有用武之地吗？可是，在别处使用训练好的网络，得先把网络的参数（就是那些variables）保存下来，怎么保存呢？其实，tensorflow已经给我们提供了很方便的API，来帮助我们实现训练参数的存储与读取，如果想了
Tensorflow入门
最新发布
qq_56246012的博客
08-25 1195
文章简单介绍了tensorflow和keras，并用鸢尾花(iris)分类案例，介绍tf.keras的基本使用流程。
tensorflow 中的保存训练的参数以及载入参数检测test集
weixin_34310369的博客
03-09 512
如何保存训练好的参数 以前面练习的一个小例子，来保存训练好的参数： 请参考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 1397
在 Tensorflow 的训练过程中，参数通过使用 Saver 类来保存和加载模型。 Saver类可以处理图中元数据和变量数据的保存和加载。我们只需要告诉 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 6474
Tensorflow教程笔记 TensorFlow 基础 TensorFlow 模型建立与训练 目录Tensorflow教程笔记模型（Model）与层（Layer）Keras 的全连接层：线性变换 + 激活函数为什么模型类是重载 `call()` 方法而不是 `__call__()` 方法？ 本章介绍如何使用 TensorFlow 快速搭建动态模型。 模型的构建： tf.keras.Model 和 tf.keras.layers 模型的损失函数： tf.keras.losses 模型的优化器： tf.k
Mac M1用tensorflow中的Keras进行基本图像分类 - TensorFlow自学笔记4
pisceshsu的博客
08-24 1445
Mac M1用tensorflow中的Keras进行基本图像分类
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 1578
深度学习框架Tensorflow学习与应用实战 1.训练模型参数(权值矩阵、偏置值)并保存到指定文件夹下 #保存训练好的模型参数(权重矩阵、偏置值等)到指定文件夹中 import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #载入数据集,one_hot:将标签转换为只有一位为1，其它...
tensorflow实例(10)--模型保存与读取，鸢尾花神经网络模型读写
雨落那秋林
02-09 2646
TensorFlow通过tf.train.Saver类实现模型的保存和提取。这样做的好处就如我们在用神经网络训练一个模型，需要非常长的时间，当把这个模型保存后，当要测试或使用这个模型时，把保存的模型提取出来，不需要再次的进行训练关于神经网络的基本原理与tensorflow的实现可参考机器学习(1)--神经网络初探 TensorFlow实例(4)--MNIST简介及手写数字分类算法本例分两段代码，分...
tensorflow创建模型时，在训练过程中对模型参数的一种保存方式。
weixin_58630610的博客
06-18 942
1、首先确定模型参数要保存的位置 checkpoint_dir = './training_checkpoints'
TensorFlow学习笔记(六) tensorflow 中的保存训练的参数以及载入参数检测test集
像风一样
04-01 1117
如何保存训练好的参数以前面练习的一个小例子，来保存训练好的参数：请参考点击打开链接import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data from tensorflow import float32 #载入数据，会自动通过一个脚本下载好数据集 mnist = input_data.rea...
TensorFlow笔记：神经网络计算与模型搭建【人工智能实践】
欠拟合则是指模型无法很好地拟合训练集和测试集，通常需要增加模型复杂度或者增加训练数据来解决。此外，还可以通过交叉验证和调参方法来优化模型的性能，提高模型的泛化能力。 总的来说，神经网络计算过程及模型...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值