3
0

parallel.rst 431 B

12345678910111213141516171819202122232425
  1. .. role:: hidden
  2. :class: hidden-section
  3. apex.parallel
  4. ===================================
  5. .. automodule:: apex.parallel
  6. .. currentmodule:: apex.parallel
  7. .. DistributedDataParallel
  8. ----------
  9. .. autoclass:: DistributedDataParallel
  10. :members:
  11. .. autoclass:: Reducer
  12. :members:
  13. .. autoclass:: SyncBatchNorm
  14. :members:
  15. Utility functions
  16. ----------------------------------
  17. .. autofunction:: convert_syncbn_model