xhj_enen的博客

tf.add_to_collection(name, value)把 value 中的变量放入到 name 作为 key 的一个集合里，也就是把多个变量统一放在一个列表中。# 在 loss 作为 key 的集合中添加变量>>> tf.add_to_collection('loss', tf.Variable(tf.truncated_normal([3, 3], stdde...

我们在上线使用一个算法模型的时候，首先必须将已经训练好的模型保存下来。tensorflow提供了两种保存模型的方式，一种是使用tf.train.Saver函数来保存TensorFlow程序的参数和完整的模型结构，保存的文件后缀为 “.ckpt”；另一种方式是将计算图保存在一个 “.pb” 文件中。使用tf.train.saver()进行模型的保存保存其中需要注意的是：变量在定义的时候必...