jettify / pytorch-optimizer

torch-optimizer -- collection of optimizers for Pytorch
Apache License 2.0
3.04k stars 300 forks source link

[Feature Request] Aida optimizer #538

Open okbalefthanded opened 9 months ago

okbalefthanded commented 9 months ago

A new optimizer that improves upon AdaBelief is proposed in this paper Aida , the optimizer was tested against SGD, Adam, AdamW, Radam, Fromage, and AdaBelief on ResNet, Transformer, and LSTM.