如何在Tensorflow中创建一个不可训练的变量？

Question

是否存在一个参数，可以将tf.Variable指定为不可训练的，使得该变量不包含在tf.trainable_variables()中？请注意保留HTML标记。

- cerebrou

Answer 1

您可以在定义中将变量标记为“不可训练”的状态，具体操作请参考TensorFlow文档。

v = tf.Variable(tf.zeros([1]), trainable=False)

这是来自于链接文档（大致相当于TensorFlow v0.11）的说明：

trainable: 如果为True，则默认情况下还会将该变量添加到GraphKeys.TRAINABLE_VARIABLES图集合中。 优化器类使用此集合作为要使用的变量的默认列表。

还有一些可以通过API更改此条件，例如tf.get_variable([v])。

- Eric Platon

---

我们可以训练输入数据吗？比如说我想在输入数据集上做与这个相反的事情。 - DuttaA

Answer 2

您可以通过两种不同的方式创建非可训练变量：

• tf.Variable(a, trainable=False)
• tf.get_variable("a", a, trainable=False)

很难将变量从可训练设置为不可训练，反之亦然。也没有简单的方法来检查变量是否可训练（您需要检查您的变量名称是否在tf.trainable_variables()列表中）。

- Salvador Dali

---

我正在从“目标检测动物园”加载一个预训练的图，但是当我运行tf.trainable_variables()时，我得到了一个空列表 - 这是什么意思？我做错了什么吗？ - Jenny

@Jenny，你弄清楚为什么目标检测动物园模型中没有可训练的变量了吗？ - lamo_738