Блог

Главная - Вопросы по программированию - как рассчитать градиенты по весам некоторых специальных слоев?

как рассчитать градиенты по весам некоторых специальных слоев?

#python #tensorflow #machine-learning #keras #deep-learning

#python #tensorflow #машинное обучение #keras #глубокое обучение

Вопрос:

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) для вычисления градиентов.Если я хочу рассчитать градиенты по весам и смещению некоторых слоев без изменения других весов, как это сделать?