lordkk的博客

原创 tf.Saver() 的使用心得

自己训练神经网络时，可能需要：1. fine-tune 预训练好的模型； 2. 分阶段训练。 这时tf.Saver()的使用必不可少。以下是自己使用这个函数的一些心得。 示例代码： import tensorflow as tf import numpy as np def add_layer(x, kin, kout, name, trainable=True): W = tf.Var...