- 博客(2)
- 资源 (1)
- 收藏
- 关注
RSS订阅原创 Tensoflow 模型持久化（下）
书接上回： 加载tensorFlow计算图时，并不需要加载全部变量。比如，有可能有一个之前训练好的五层神经网络模型，但现在想尝试一个六层的神经网络，那么可以将前面五层神经网络中的参数直接加载到新的模型，而仅仅将最后一层神经网络重新训练。 为了保存或者加载部分变量，在声明tf.train.Saver类时可以提供一个列表来制定需要保存或者加载的变量，比如在加载模型的代码中使用...
2018-05-12 15:14:13
315
1
原创 TensorFlow模型持久化（上）
Tensorflow 提供了一个非常简单的API 来保存和还原一个神经网络模型这个API就是tf.train.Saver类，一下代码给出了保存Tensorflow计算图的方import tensorflow as tf#声明两个变量并计算他们的和 v1 = tf.Variable(tf.constant(1.0, shape = [1]), name = "v1") v2 = tf.Var...
2018-05-12 11:53:00
442
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人