3
0

optimizers.rst 356 B

1234567891011121314151617181920212223
  1. .. role:: hidden
  2. :class: hidden-section
  3. apex.optimizers
  4. ===================================
  5. .. automodule:: apex.optimizers
  6. .. currentmodule:: apex.optimizers
  7. .. FusedAdam
  8. ----------
  9. .. autoclass:: FusedAdam
  10. :members:
  11. .. autoclass:: FusedLAMB
  12. :members:
  13. .. autoclass:: FusedNovoGrad
  14. :members:
  15. .. autoclass:: FusedSGD
  16. :members: