前面的两篇博文 第一篇:简单的模型保存和加载,会包含所有的信息:神经网络的op,node,args等; 第二篇:选择性的进行模型参数的保存与加载。 本篇介绍,只保存和加载神经网络的计算图,即前向传播的过程。 源码路径:https://github.com/suonikeyinsuxiao/tf_no Read More
上一遍博文提到 有些场景下,可能只需要保存或加载部分变量,并不是所有隐藏层的参数都需要重新训练。 在实例化tf.train.Saver对象时,可以提供一个列表或字典来指定需要保存或加载的变量。 执行上面的代码,会报错:tensorflow.python.framework.errors_impl.F Read More