From 182f604d323670c1ac8a1780945ebd0c95e2ed1c Mon Sep 17 00:00:00 2001 From: Antoni Baum Date: Tue, 14 Jun 2022 00:20:24 +0200 Subject: [PATCH] [docs] Fix bad argument name in PTL docs (#25736) Fixes bad argument name in PTL docs. This is just a quick fix - we should be testing the code snippet. --- .../ray-core/examples/using-ray-with-pytorch-lightning.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc/source/ray-core/examples/using-ray-with-pytorch-lightning.rst b/doc/source/ray-core/examples/using-ray-with-pytorch-lightning.rst index 9cfd9f8f7..182a14d9b 100644 --- a/doc/source/ray-core/examples/using-ray-with-pytorch-lightning.rst +++ b/doc/source/ray-core/examples/using-ray-with-pytorch-lightning.rst @@ -57,7 +57,7 @@ Here is an example of using the ``RayPlugin`` for Distributed Data Parallel trai # Create your PyTorch Lightning model here. ptl_model = MNISTClassifier(...) - plugin = RayPlugin(num_workers=4, cpus_per_worker=1, use_gpu=True) + plugin = RayPlugin(num_workers=4, num_cpus_per_worker=1, use_gpu=True) # If using GPUs, set the ``gpus`` arg to a value > 0. # The actual number of GPUs is determined by ``num_workers``.