wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-17 12:40:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update hooks.md

Browse Source
This commit is contained in:
William Falcon
2019-07-28 08:02:29 -04:00
committed by GitHub
parent 5a6ee935f0
commit 579f111637
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/Trainer/hooks.md
+4 -5
View File
@@ -4,11 +4,10 @@
There are cases when you might want to do something different at different parts of the training/validation loop. 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. 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: **Contributing** If there's a hook you'd like to add, simply:
1. Fork PytorchLightning. 1. Fork PytorchLightning.
2. Add the hook [here](https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/root_module/hooks.py). 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) 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.
where it should be called.
--- ---
#### on_epoch_start #### on_epoch_start