有人有 L-BFGS 算法的 Tensorflow 2 tf.keras 子类吗?如果想使用 L-BFGS,目前有两个(官方)选项:
- TF 概率
- SciPy 优化
这两个选项使用起来相当麻烦,尤其是在使用自定义模型时。所以我计划实现 tf.keras.optimizers 的自定义子类以使用 L-BFGS。但在我开始之前,我很好奇,是否有人已经完成了这项任务?
最佳答案
我已经实现了 keras 和 SciPy 优化之间的接口(interface)。 https://github.com/pedro-r-marques/keras-opt
我默认使用“cg”,但您也应该能够使用“l-bfgs”。查看单元测试以了解示例用法。我会尽快添加文档。
关于tensorflow - 是否有 L-BFGS 的 tf.keras.optimizers 实现?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63882987/