如何获取tf.layers的变量列表

时间:2018-01-26 05:28:44

标签: python tensorflow conv-neural-network

我想在一个图层的指定变量上应用渐变。因此需要变量列表作为参数传递:var_list到optimizer.minimize。但我不知道如何获取它们。

如:

LocalDate

我只想训练核心变量,重量和偏差  图层b ,并保持图层不受影响。

我该怎么做或者我应该使用较低级别来实现它?

1 个答案:

答案 0 :(得分:0)

嗯,你想训练layer b的所有参数(只有重量和偏差),但想保持layer a的参数不变(正如你所描述的那样),那么你可以将trainable=False参数传递给tf.layers.conv2d

但是如果你想要更多地控制变量,你可以用tf.trainable_variables()打印变量后手动选择要训练的变量