Tensorflow学习：saver_restore模型载入（手写数字识别，简单神经网络）

最新推荐文章于 2021-09-27 16:19:50 发布

空城不空99

最新推荐文章于 2021-09-27 16:19:50 发布

阅读量278

点赞数 1

分类专栏： Tensorflow学习文章标签： tensorflow 深度学习神经网络机器学习

本文链接：https://blog.csdn.net/cui1004/article/details/107817927

版权

Tensorflow学习：saver_restore模型载入（手写数字识别，简单神经网络）

  saver = tf.train.Saver()
  saver.restore(sess,'net/my_net.ckpt')

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data

# 载入数据集
mnist = input_data.read_data_sets("MNST_data", one_hot=True)

# 每个批次的大小：一次性向神经网络中放入100张图片进行训练:以矩阵的形式放进去
batch_size = 100
# 计算一共多少个批次 //:整除
n_batch = mnist.train.num_examples // batch_size

# 定义两个placeholder  None:过会以一个批次喂进去，784：28*28(把一个图片拉成一个784的向量) 10:输出时0-9的10个数
x = tf