上传者: 42129412
|
上传时间: 2021-11-08 15:53:52
|
文件大小: 17KB
|
文件类型: -
PyTorch最小化
的包装,使其成为PyTorch优化器,在PyTorch中实现共轭梯度,BFGS,l-BFGS,SLSQP,牛顿共轭梯度,信任区域方法等。
警告:尽管(仅此而已)足够小以至于易于阅读,但它只是一个概念证明,并不一定是可靠的。
快速开始
安装
依存关系:
pytorch
scipy
以下安装过程将不会检查它们是否已安装。
这个软件包可以直接从Github用pip安装:
python -m pip install git+https://github.com/gngdb/pytorch-minimize.git
或通过克隆存储库然后安装:
git clone https://github.com/gngdb/pytorch-minimize.git
cd pytorch-minimize
python -m pip install .
使用优化器
优化器类是