我使用tf.estimator
实现我的模型,并且效果很好:
model = tf.estimator.Estimator(...)
for _ in range(epochs):
model.train(train_input_fn, steps=STEPS_PER_EPOCH)
model.evaluate(eval_test_fn, name='test')
现在,我想根据评估指标(即准确性,验证损失等)来调整学习率,这类似于keras中的reducelronplateau
。
我该怎么做?