我想在一个图层的指定变量上应用渐变。因此需要变量列表作为参数传递:var_list到optimizer.minimize。但我不知道如何获取它们。
如:
LocalDate
我只想训练核心变量,重量和偏差 图层b ,并保持图层不受影响。
我该怎么做或者我应该使用较低级别来实现它?
答案 0 :(得分:0)
嗯,你想训练layer b
的所有参数(只有重量和偏差),但想保持layer a
的参数不变(正如你所描述的那样),那么你可以将trainable=False
参数传递给tf.layers.conv2d
。
但是如果你想要更多地控制变量,你可以用tf.trainable_variables()
打印变量后手动选择要训练的变量