ColossalAI/docs/colossalai/colossalai.zero.sharded_mod...

6 lines
163 B
ReStructuredText

colossalai.zero.sharded\_model.sharded\_grad
============================================
.. automodule:: colossalai.zero.sharded_model.sharded_grad
:members: