Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
ai-forever
GitHub Repository: ai-forever/sber-swap
Path: blob/main/apex/docs/source/parallel.rst
1778 views
.. role:: hidden
    :class: hidden-section

apex.parallel
===================================

.. automodule:: apex.parallel
.. currentmodule:: apex.parallel

.. DistributedDataParallel
   ----------

.. autoclass:: DistributedDataParallel
    :members:

.. autoclass:: Reducer
    :members:

.. autoclass:: SyncBatchNorm
    :members:

Utility functions
----------------------------------

.. autofunction:: convert_syncbn_model