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

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

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

文章目录

    • 一、保存:save
    • 二、提取:restore
    • 三、读取 checkpoint 中的 tensor
    • 四、`tf.train.latest_checkpoint()`

一、保存:save

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

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

你可能感兴趣的:(TensorFlow,学习,深度学习知识点笔记,TensorFlow,中,Saver,保存读取)