ColossalAI/docs/colossalai/colossalai.nn.optimizer.fus...

6 lines
137 B
ReStructuredText

colossalai.nn.optimizer.fused\_lamb
===================================
.. automodule:: colossalai.nn.optimizer.fused_lamb
:members: