Tensorflow学习笔记10-Saver保存读取

原创

于 2020-12-03 20:05:47 发布 · 240 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #tensorflow #深度学习 #numpy #神经网络

这篇博客介绍了如何使用TensorFlow进行模型的保存和提取。首先，导入必要的模块并建立神经网络的权重W和偏置b，然后通过tf.train.Saver()创建一个保存器，将模型保存到'my_net/save_net.ckpt'目录。在提取模型时，创建临时的W和b变量，找到保存的文件路径，并利用saver.restore()恢复模型的参数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

保存

import所需的模块, 然后建立神经网络当中的 W 和 b, 并初始化变量。
存时, 首先要建立一个 tf.train.Saver() 用来保存, 提取变量. 再创建一个名为my_net的文件夹, 用这个 saver 来保存变量到这个目录 “my_net/save_net.ckpt”.

#---------------保存到文件-------------#

W = tf.Variable([[1,2,3],[3,4,5]], dtype=tf.float32, name='weight')
b = tf.Variable([

最低0.47元/天解锁文章

200万优质内容无限畅学