tensorflow学习（4）之saver保存读取

最新推荐文章于 2021-03-08 14:38:52 发布

onejuliar

最新推荐文章于 2021-03-08 14:38:52 发布

阅读量248

点赞数 2

分类专栏： TensorFlow

本文链接：https://blog.csdn.net/juliarjuliar/article/details/89244527

版权

本文介绍了如何使用TensorFlow中的Saver模块来保存和加载神经网络模型。通过示例代码展示了具体的操作步骤，包括建立Saver、指定保存路径以及加载模型的详细过程。

摘要由CSDN通过智能技术生成

莫凡2017tensorflow（使用更简便版）https://github.com/MorvanZhou/Tensorflow-Tutorial

怎样用 Tensorflow 中的 saver 保存和加载

https://morvanzhou.github.io/tutorials/machine-learning/tensorflow/5-06-save/

如果保存搭建好的神经网络：

1.保存

import所需的模块, 然后建立神经网络当中的 W 和 b, 并初始化变量.

import tensorflow as tf
import numpy as np

## Save to file
# remember to define the same dtype and shape when restore
W = tf.Variable([[1,2,3],[3,4,5]], dtype=tf.float32, name='weights')
b = tf.Variable([[1,2,3]], dtype=tf.float32, name='biases')

# init= tf.initialize_all_variables() # tf 马上就要废弃这种写法
# 替换成下面的写法:
init = tf.global_variables_initializer()

保存时, 首先要建立一个