如何从tensorflow上的txt文件加载变量?

时间:2019-03-14 09:51:22

标签: python tensorflow

我是tensorflow的初学者。

我将训练输出的权重和偏差存储在parameter.txt中。

并通过“ quantization.py”对“ parameter.txt”进行量化。

我想将“ quantize_parameter.txt”中的权重和偏差设置为原始训练中的权重和偏差的初始值,然后再次进行训练。

但是我不知道如何将txt文件的内容加载到变量中。

部分培训代码:

def init_layer(self, name, n_inputs, n_outputs):
    W = tf.get_variable(name, shape=[
                        n_inputs, n_outputs], initializer=tf.contrib.layers.xavier_initializer())
    return W


W_1 = self.init_layer('W_1', 784, 512)
fc1 = tf.matmul(self.input, W_1)
fc1 = tf.nn.dropout(tf.nn.relu(fc1), self.keep_prob)

如何使“变量”从txt文件接收值?

谢谢。

0 个答案:

没有答案