如何在Tensorflow中冻结基本RNN单元的权重

时间:2018-08-08 23:35:09

标签: python tensorflow rnn

如何在Tensorflow中为BasicRNNCell编写子类方法,以便我可以修改权重矩阵的少量权重并将其冻结并训练模型。

    basic_cell= tf.contrib.rnn.BasicRNNCell(num_units=n_neurons)
    outputs, states = tf.nn.dynamic_rnn(basic_cell, X, dtype=tf.float32)

具有权重和偏见。现在我该如何更改这些权重的几个条目,冻结它们并训练模型?

0 个答案:

没有答案