当我使用 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)
模型训练时如何从优化器中检索当前学习率?
如果您能提供任何帮助,我将不胜感激。:)