May, 2020

Geoopt: PyTorch 中的黎曼优化

TL;DRGeoopt 是一个用于基于黎曼流形优化的,面向研究的模块化开源软件包,支持基本的黎曼随机梯度下降以及自适应优化算法,并提供多种算法和计算方法,可实现在现有模型中集成的几何感知神经网络层。