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

标签 tensorflow optimization keras subclassing

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

  1. TF 概率
  2. 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/

相关文章:

python - model.predict 导致 oom 问题,但 model.fit 不会 :

tensorflow - Tensorflow 嵌入层内部的网络结构是什么?

tensorflow - 如何将签名名称设置为serving_default以避免部署到GCP AI Platform后出现错误

python - 如何将多线程添加到连续的 Python 函数中?

python - 如何使用 keras 和 tensorflow 后端将密集层的输出作为 numpy 数组?

python - 图像生成器缺少 unet keras 的位置参数

python - 绘制 Keras 单个时期内的损失演变

python - 为什么我的 tensorflow 模型输出在 x 个时期后变成 NaN?

c++ - (-1)^n 在 C++ 中求幂

c - 数字的安全 gcc 优化选项