From 579f111637498056154747a0e9b131ab1b69cde9 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sun, 28 Jul 2019 08:02:29 -0400 Subject: [PATCH] Update hooks.md --- docs/Trainer/hooks.md | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) 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