如何强制scipy.optimize.fmin_l_bfgs_b使用'dtype = float32'

Question

我正在尝试使用Python中的GPU计算来优化功能，因此我更喜欢使用dtype dtype=float32将所有数据存储为ndarrays。

当我使用scipy.optimize.fmin_l_bfgs_b ，我注意到即使我将float32 ndarray作为初始搜索点x0 ，优化器也始终将float64 （在我的64位计算机上）参数传递给目标函数和渐变函数。 当我使用cg优化器scipy.optimize.fmin_cg ，这是不同的，其中当我将float32数组作为x0传递时，优化器将在所有随后的目标/梯度函数调用中使用float32 。

所以我的问题是：我可以强制scipy.optimize.fmin_l_bfgs_b来优化float32参数，例如scipy.optimize.fmin_cg吗？

谢谢！

Answer 1

我不确定您能做到。 fmin_l_bfgd_b不是由纯python代码提供的，而是由扩展名（FORTRAN代码的包装）提供的。 在Win32 / 64平台上，可以在\\scipy\\optimize\\_lbfgsb.pyd找到它。 仅当您可以以不同方式编译扩展或修改FORTRAN代码时，才可能需要您想要的东西。 如果您检查该FORTRAN代码，则它在整个位置都具有double precision ，这基本上是float64 。 我不确定只是将它们全部更改为single precision 。

在其他优化方法中，FORTRAN还提供了cobyla 。 鲍威尔的方法也是。