Skip to content

Commit

Permalink
[2.7] Cleanup all LightningTrainer Mentions in Ray Doc (ray-project#3…
Browse files Browse the repository at this point in the history
…9406)

Signed-off-by: woshiyyya <xiaoyunxuan1998@gmail.com>
Signed-off-by: Jim Thompson <jimthompson5802@gmail.com>
  • Loading branch information
woshiyyya authored and jimthompson5802 committed Sep 12, 2023
1 parent b5228c7 commit bd25d78
Show file tree
Hide file tree
Showing 5 changed files with 970 additions and 192 deletions.
2 changes: 1 addition & 1 deletion doc/source/ray-overview/examples.rst
Original file line number Diff line number Diff line change
Expand Up @@ -1373,7 +1373,7 @@ Ray Examples
:link: /train/examples/lightning/vicuna_13b_lightning_deepspeed_finetune
:link-type: doc

Fine-tune vicuna-13b-v1.3 with DeepSpeed and LightningTrainer
Fine-tune vicuna-13b-v1.3 with DeepSpeed, PyTorch Lightning and Ray Train

.. grid-item-card:: :bdg-secondary:`Code example`
:class-item: gallery-item training llm pytorch nlp
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@
"source": [
"## Prepare Dataset and Module\n",
"\n",
"The Pytorch Lightning Trainer takes either `torch.utils.data.DataLoader` or `pl.LightningDataModule` as data inputs. You can keep using them without any changes for the Ray AIR LightningTrainer. "
"The Pytorch Lightning Trainer takes either `torch.utils.data.DataLoader` or `pl.LightningDataModule` as data inputs. You can keep using them without any changes with Ray Train. "
]
},
{
Expand Down
5 changes: 0 additions & 5 deletions doc/source/tune/examples/includes/mnist_ptl_mini.rst
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,4 @@
MNIST PyTorch Lightning Example
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. note::

In version 2.4, we introduced :class:`LightningTrainer <ray.train.lightning.LightningTrainer>`, which provides better integration with PyTorch Lightning.
For more information, please refer to :ref:`Using PyTorch Lightning with Tune <tune-pytorch-lightning-ref>`.

.. literalinclude:: /../../python/ray/tune/examples/mnist_ptl_mini.py

0 comments on commit bd25d78

Please sign in to comment.