Module burn::optim

Expand description

Optimizer module.

Modules§

  • Adaptor module for optimizers.
  • Weight decay module for optimizers.
  • Momentum module for optimizers.
  • Record module for optimizers.

Structs§

Traits§

  • General trait to optimize module.
  • Simple optimizer is an opinionated trait to simplify the process of implementing an optimizer.