Skip to content

Commit

Permalink
Document ddp_find_unused_parameters_true in Fabric (#19564)
Browse files Browse the repository at this point in the history
  • Loading branch information
leng-yue committed Mar 4, 2024
1 parent 48c39ce commit 34f0369
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion docs/source-fabric/api/fabric_args.rst
Original file line number Diff line number Diff line change
Expand Up @@ -36,13 +36,16 @@ See also: :doc:`../fundamentals/accelerators`
strategy
========

Choose a training strategy: ``"dp"``, ``"ddp"``, ``"ddp_spawn"``, ``"xla"``, ``"deepspeed"``, ``"fsdp"````.
Choose a training strategy: ``"dp"``, ``"ddp"``, ``"ddp_spawn"``, ``"ddp_find_unused_parameters_true"``, ``"xla"``, ``"deepspeed"``, ``"fsdp"``.

.. code-block:: python
# Running with the DistributedDataParallel strategy on 4 GPUs
fabric = Fabric(strategy="ddp", accelerator="gpu", devices=4)
# Running with the DDP strategy with find unused parameters enabled on 4 GPUs
fabric = Fabric(strategy="ddp_find_unused_parameters_true", accelerator="gpu", devices=4)
# Running with the DDP Spawn strategy using 4 CPU processes
fabric = Fabric(strategy="ddp_spawn", accelerator="cpu", devices=4)
Expand Down

0 comments on commit 34f0369

Please sign in to comment.