diff --git a/docs/Trainer/index.md b/docs/Trainer/index.md index 7731e4e4b5..88779797d0 100644 --- a/docs/Trainer/index.md +++ b/docs/Trainer/index.md @@ -1,5 +1,5 @@ # Trainer -[[Github Code](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/trainer.py)] +[[Github Code](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/trainer/trainer.py)] The lightning trainer abstracts best practices for running a training, val, test routine. It calls parts of your model when it wants to hand over full control and otherwise makes training assumptions which are now standard practice in AI research.