L-BFGS是否有tf.keras.optimizers实现?

问题描述

有人为L-BFGS算法提供Tensorflow 2 tf.keras子类吗?如果要使用L-BFGS,则目前有两个(官方)选项:

  1. TF概率
  2. SciPy优化

这两个选项使用起来非常麻烦,尤其是在使用自定义模型时。因此,我计划实现tf.keras.optimizers的自定义子类以使用L-BFGS。但是在开始之前,我很好奇,是否有人已经完成了这项任务?

解决方法

我已经实现了keras和SciPy优化之间的接口。 https://github.com/pedro-r-marques/keras-opt

默认情况下,我使用的是“ cg”,但是您也应该能够使用“ l-bfgs”。看一下单元测试的用法示例。我将尽快添加文档。