diff --git a/docs/index.md b/docs/index.md index 03b500d441..8a7c63c36e 100644 --- a/docs/index.md +++ b/docs/index.md @@ -2,8 +2,10 @@ To start a new project define these two files. 1. [Define a LightningModule](/pytorch-lightning/LightningModule/RequiredTrainerInterface/) -2. Pick a trainer +2. [Define a trainer](https://williamfalcon.github.io/pytorch-lightning/Trainer/) + - Examples: - [Basic CPU Trainer](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/examples/new_project_templates/single_cpu_template.py) + - [Multi-GPU Trainer](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/examples/new_project_templates/single_gpu_node_template.py) - [GPU cluster Trainer](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/examples/new_project_templates/multi_node_cluster_template.py) ###### Docs shortcuts