pytorch_optimizer icon indicating copy to clipboard operation
pytorch_optimizer copied to clipboard

Plans for pytorch_optimizer v3

Open kozistr opened this issue 2 years ago • 1 comments

In pytorch-optimizer v3, loss function will be added. So, finally, the optimizer & lr scheduler & loss function are all in one package.

Feature

  • [x] support at least 60 optimizers
  • [x] support at least 10 objectives
  • [x] support bitsandbytes (& 4-bit optimizers)

Refactor

  • [x] Organize utils

Docs

  • [x] Organize documentation
  • [x] Support contribution guide (implementation, test, etc...)
  • [x] Add issue templates
  • [x] Migrate to mkdocs
  • [x] Create Q&A page
  • [ ] Benchmark on ImageNet

Test

  • [x] Organize test cases

kozistr avatar May 09 '23 03:05 kozistr

IMAE for Noise-Robust Learning: Mean Absolute Error Does Not Treat Examples Equally and Gradient Magnitude's Variance Matters https://arxiv.org/abs/1903.12141

redknightlois avatar May 24 '23 20:05 redknightlois