Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
optimizers.rst 356 Bytes
Newer Older
Michael Carilli's avatar
Updating documentation for merged utilities  
Michael Carilli committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14 
.. role:: hidden
    :class: hidden-section

apex.optimizers
===================================

.. automodule:: apex.optimizers
.. currentmodule:: apex.optimizers

.. FusedAdam
   ----------

.. autoclass:: FusedAdam
    :members:
Michael Carilli's avatar
Updating docstrings for fused optimizers  
Michael Carilli committed
15
16
17
18
19
20
21
22
23 

.. autoclass:: FusedLAMB
    :members:

.. autoclass:: FusedNovoGrad
    :members:

.. autoclass:: FusedSGD
    :members: