TensorFlow 中的 save、restore 以及从 checkpoint 中读取变量

南淮北安

于 2019-09-17 19:33:46 发布

阅读量741

点赞数

分类专栏： TensorFlow 学习深度学习知识点笔记文章标签： TensorFlow 中 Saver 保存读取

本文链接：https://blog.youkuaiyun.com/nanhuaibeian/article/details/100936416

版权

84 篇文章 ¥39.90 ¥99.00

订阅专栏

56 篇文章

订阅专栏

本文介绍了如何使用TensorFlow的Saver模块进行模型的保存和恢复，包括保存神经网络参数为.ckpt文件，以及从checkpoint中读取变量，详细解析了保存、提取和检查点文件的读取过程。

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

我们搭建好一个神经网络，用大量的数据训练好之后，肯定希望保存神经网络里面的参数，用于下次加载。那我们该怎么做呢？

TensorFlow为我们提供了Saver来保存和加载神经网络的参数

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

import tensorflow as tf
import numpy as np
# 存储时，需要记得定义 dtype 和 shape 选项
W = tf.Variable([[<

了解本专栏