学习率取决于评估

时间:2019-03-16 08:38:35

标签: tensorflow tensorflow-estimator

我使用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

我该怎么做?

0 个答案:

没有答案