pytorch-minimize:使用PyTorch进行牛顿和拟牛顿优化

时间:2024-05-01 21:19:16
【文件属性】:

文件名称:pytorch-minimize:使用PyTorch进行牛顿和拟牛顿优化

文件大小:54KB

文件格式:ZIP

更新时间:2024-05-01 21:19:16

optimization minimization pytorch newton-method quasi-newton-method

PyTorch最小化 Pytorch-minimize表示一组实用程序,用于最小化PyTorch中一个或多个变量的标量函数。 它受到SciPy的optimize模块和MATLAB的极大启发。 与SciPy和MATLAB不同,无需提供jacobian和hessian函数来进行pytorch最小化求解器,并且永远不会使用数值近似。 CPU和CUDA均受支持。 目前,仅实现了不受约束的最小化例程。 作者:鲁本·费曼(Reuben Feinman) 乍看上去: import torch from fmin import minimize def rosen ( x ): return torch . sum ( 100 * ( x [..., 1 :] - x [..., : - 1 ] ** 2 ) ** 2 + ( 1 - x [...


【文件预览】:
pytorch-minimize-master
----fmin()
--------minimize.py(3KB)
--------bfgs.py(9KB)
--------__init__.py(156B)
--------line_search.py(2KB)
--------trust_constr.py(5KB)
--------newton.py(15KB)
----examples()
--------scipy_benchmark.py(3KB)
--------constrained_optimization_adversarial_examples.ipynb(13KB)
--------rosen_minimize.ipynb(57KB)
----README.md(5KB)

网友评论