如何针对某些特殊图层的权重计算梯度?

时间:2019-03-27 00:46:36

标签: python tensorflow machine-learning keras deep-learning

TensorFlow使用以下方法计算梯度(https://www.tensorflow.org/tutorials/eager/custom_training_walkthrough

def grad(model, inputs, targets):
  with tf.GradientTape() as tape:
    loss_value = loss(model, inputs, targets)
  return loss_value, tape.gradient(loss_value, model.trainable_variables)

模型使用模型(model.trainable_variables)中的所有变量来计算梯度。如果我想针对某些层权重计算梯度并在不更改其他权重的情况下进行偏置,怎么办?

0 个答案:

没有答案
相关问题