python - pytorch：实现自定义优化器

Question

我正在尝试使用 pytorch 实现稍微不同的 SGD 版本，并在一些数据集上对其进行测试。我需要编写一个自定义优化器来训练我的模型，但是我找不到任何解释如何这样做的指南。有人能告诉我如何继续或建议我任何有用的参考吗？

提前致谢。