torch-optimizer -- Pytorch的优化器集合-python

【文件属性】：

文件名称：torch-optimizer -- Pytorch的优化器集合-python

文件大小：34.24MB

文件格式：ZIP

更新时间：2024-07-08 13:04:37

机器学习

torch-optimizer -- collection of optimizers for Pytorch torch-optimizer torch-optimizer -- 与 optim 模块兼容的 PyTorch 优化器集合。 简单示例 import torch_optimizer as optim # model = ... optimizer = optim.DiffGrad(model.parameters(), lr=0.001) optimizer.step() 安装 安装过程很简单，只需：$ pip install torch_optimizer 文档 https:// pytorch-optimizer.rtfd.io 支持的优化器 AccSGD https://arxiv.org/abs/1803.05591 AdaBound https://arxiv.org/abs/1902.09843 AdaMod https://arxiv.org/abs/1910.12249 DiffGrad https:// arxiv.org/abs/1909.11015