diff --git a/docs/Trainer/hooks.md b/docs/Trainer/hooks.md index 061ffa1a..c7a4bfbe 100644 --- a/docs/Trainer/hooks.md +++ b/docs/Trainer/hooks.md @@ -4,11 +4,10 @@ There are cases when you might want to do something different at different parts of the training/validation loop. To enable a hook, simply override the method in your LightningModule and the trainer will call it at the correct time. -**Contributing** If there's a hook you'd like to add, simply: -1. Fork PytorchLightning. -2. Add the hook [here](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/root_module/hooks.py). -3. Add the correct place in the [Trainer](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/trainer.py) -where it should be called. +**Contributing** If there's a hook you'd like to add, simply: +1. Fork PytorchLightning. +2. Add the hook [here](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/root_module/hooks.py). +3. Add the correct place in the [Trainer](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/trainer.py) where it should be called. --- #### on_epoch_start