mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
cd82096973
Fixes #155017 Pull Request resolved: https://github.com/pytorch/pytorch/pull/155298 Approved by: https://github.com/svekars Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
24 lines
647 B
Markdown
24 lines
647 B
Markdown
```{role} hidden
|
|
---
|
|
class: hidden-section
|
|
---
|
|
|
|
```
|
|
# Generic Join Context Manager
|
|
|
|
The generic join context manager facilitates distributed training on uneven
|
|
inputs. This page outlines the API of the relevant classes: {class}`Join`,
|
|
{class}`Joinable`, and {class}`JoinHook`. For a tutorial, see
|
|
[Distributed Training with Uneven Inputs Using the Join Context Manager](https://pytorch.org/tutorials/advanced/generic_join.html).
|
|
|
|
```{eval-rst}
|
|
.. autoclass:: torch.distributed.algorithms.Join
|
|
:members:
|
|
|
|
.. autoclass:: torch.distributed.algorithms.Joinable
|
|
:members:
|
|
|
|
.. autoclass:: torch.distributed.algorithms.JoinHook
|
|
:members:
|
|
|
|
``` |