文件名称: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)