gpt4 book ai didi

tensorflow - 使用 tf.trainable_variables() 显示可训练变量的名称

转载 作者:行者123 更新时间:2023-12-03 00:12:53 27 4
gpt4 key购买 nike

我正在 RNNCell 中处理权重

我有以下代码

var_names = []    
for var in tf.trainable_variables():
var_names.append(var.name)

在另一个名为 model.py 的文件中,我打印了刚刚附加到 var_names 的可训练变量的名称。但是,我发现可训练变量的“名称”属性没有用,因为描述性不强。

RNNcell 的权重有名称吗?

如果这有用,这里是另一段可能相关的代码:

cell_fn = tf.nn.rnn_cell.GRUCell   
rnn_fw_1 = cell_fn(num_hidden_1, **additional_cell_args)
rnn_fw_1 = tf.nn.rnn_cell.DropoutWrapper(rnn_fw_1, input_keep_prob=keep_prob_1)

最佳答案

试试这个:

variables_names = [v.name for v in tf.trainable_variables()]
values = sess.run(variables_names)
for k, v in zip(variables_names, values):
print "Variable: ", k
print "Shape: ", v.shape
print v

关于tensorflow - 使用 tf.trainable_variables() 显示可训练变量的名称,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41951657/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com