hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[docs] Fix bad argument name in PTL docs (#25736)

Browse source 
Fixes bad argument name in PTL docs. This is just a quick fix - we should be testing the code snippet.
This commit is contained in:
Antoni Baum 2022-06-14 00:20:24 +02:00 • committed by GitHub
parent fde61a77be
commit 182f604d32
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
doc/source/ray-core/examples/using-ray-with-pytorch-lightning.rst
View file

@ -57,7 +57,7 @@ Here is an example of using the ``RayPlugin`` for Distributed Data Parallel trai
# Create your PyTorch Lightning model here. # Create your PyTorch Lightning model here.
ptl_model = MNISTClassifier(...) 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. # If using GPUs, set the ``gpus`` arg to a value > 0.
# The actual number of GPUs is determined by ``num_workers``. # The actual number of GPUs is determined by ``num_workers``.