问题描述
有人为L-BFGS算法提供Tensorflow 2 tf.keras子类吗?如果要使用L-BFGS,则目前有两个(官方)选项:
- TF概率
- SciPy优化
这两个选项使用起来非常麻烦,尤其是在使用自定义模型时。因此,我计划实现tf.keras.optimizers的自定义子类以使用L-BFGS。但是在开始之前,我很好奇,是否有人已经完成了这项任务?
解决方法
我已经实现了keras和SciPy优化之间的接口。 https://github.com/pedro-r-marques/keras-opt
默认情况下,我使用的是“ cg”,但是您也应该能够使用“ l-bfgs”。看一下单元测试的用法示例。我将尽快添加文档。