我是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文件接收值?
谢谢。