mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
Deployed 579f111 with MkDocs version: 1.0.4
This commit is contained in:
@@ -638,11 +638,10 @@
|
||||
<p>[<a href="https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/root_module/hooks.py">Github Code</a>] </p>
|
||||
<p>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.</p>
|
||||
<p><strong>Contributing</strong> If there's a hook you'd like to add, simply:
|
||||
1. Fork PytorchLightning.
|
||||
2. Add the hook <a href="https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/root_module/hooks.py">here</a>. <br />
|
||||
3. Add the correct place in the <a href="https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/trainer.py">Trainer</a>
|
||||
where it should be called.</p>
|
||||
<p><strong>Contributing</strong> If there's a hook you'd like to add, simply: <br />
|
||||
1. Fork PytorchLightning. <br />
|
||||
2. Add the hook <a href="https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/root_module/hooks.py">here</a>. <br />
|
||||
3. Add the correct place in the <a href="https://github.com/williamFalcon/pytorch-lightning/blob/master/pytorch_lightning/models/trainer.py">Trainer</a> where it should be called. </p>
|
||||
<hr />
|
||||
<h4 id="on_epoch_start">on_epoch_start</h4>
|
||||
<p>Called in the training loop at the very beginning of the epoch. </p>
|
||||
|
||||
Binary file not shown.
Reference in New Issue
Block a user