小编yun*_*yun的帖子

tensorflow 2.0 自定义训练循环的学习率

当我使用 tensorflow 2.0 自定义训练循环时,是否有任何函数或方法可以显示学习率?

这是张量流指南的示例:

def train_step(images, labels):
  with tf.GradientTape() as tape:
    predictions = model(images)
    loss = loss_object(labels, predictions)
  gradients = tape.gradient(loss, model.trainable_variables)
  optimizer.apply_gradients(zip(gradients, model.trainable_variables))

  train_loss(loss)
  train_accuracy(labels, predictions)
Run Code Online (Sandbox Code Playgroud)

模型训练时如何从优化器中检索当前学习率?

如果您能提供任何帮助,我将不胜感激。:)

python tensorflow

8
推荐指数
2
解决办法
2739
查看次数

标签 统计

python ×1

tensorflow ×1