更新一些嵌入词

时间:2017-08-17 20:46:13

标签: tensorflow

我在https://github.com/dennybritz/cnn-text-classification-tf中试验代码并使用Glove初始化嵌入词。现在我想冻结大部分嵌入词,只训练一小部分。我能找到的唯一相关主题是Update only part of the word embedding matrix in Tensorflow

但是,在我的情况下,我仍然无法理解如何正确地做到这一点。鉴于这样的事情:

optimizer = tf.train.AdamOptimizer(1e-3)
grads_and_vars = optimizer.compute_gradients(cnn.loss)
train_op = optimizer.apply_gradients(grads_and_vars)

在这种情况下,如何更改索引以仅训练部分嵌入?

谢谢!

0 个答案:

没有答案