2020-02-27 22:21:51 +00:00
|
|
|
Hooks
|
|
|
|
|
-----
|
|
|
|
|
|
2020-02-27 12:35:20 +00:00
|
|
|
.. automodule:: pytorch_lightning.core.hooks
|
|
|
|
|
|
|
|
|
|
Full list of hooks
|
2020-02-27 22:21:51 +00:00
|
|
|
|
2020-02-11 04:55:22 +00:00
|
|
|
|
|
|
|
|
Training set-up
|
2020-02-27 22:21:51 +00:00
|
|
|
================
|
2020-02-11 04:55:22 +00:00
|
|
|
- init_ddp_connection
|
|
|
|
|
- init_optimizers
|
|
|
|
|
- configure_apex
|
|
|
|
|
- configure_ddp
|
2020-02-25 03:23:25 +00:00
|
|
|
- train_dataloader
|
|
|
|
|
- test_dataloaders
|
|
|
|
|
- val_dataloaders
|
2020-02-11 04:55:22 +00:00
|
|
|
- summarize
|
|
|
|
|
- restore_weights
|
|
|
|
|
|
|
|
|
|
Training loop
|
2020-02-27 12:35:20 +00:00
|
|
|
=============
|
2020-02-11 04:55:22 +00:00
|
|
|
|
|
|
|
|
- on_epoch_start
|
|
|
|
|
- on_batch_start
|
|
|
|
|
- tbptt_split_batch
|
|
|
|
|
- training_step
|
|
|
|
|
- training_end (optional)
|
|
|
|
|
- backward
|
|
|
|
|
- on_after_backward
|
|
|
|
|
- optimizer.step()
|
|
|
|
|
- on_batch_end
|
|
|
|
|
- on_epoch_end
|
|
|
|
|
|
|
|
|
|
Validation loop
|
2020-02-27 12:35:20 +00:00
|
|
|
===============
|
2020-02-11 04:55:22 +00:00
|
|
|
|
|
|
|
|
- model.zero_grad()
|
|
|
|
|
- model.eval()
|
|
|
|
|
- torch.set_grad_enabled(False)
|
|
|
|
|
- validation_step
|
|
|
|
|
- validation_end
|
|
|
|
|
- model.train()
|
|
|
|
|
- torch.set_grad_enabled(True)
|
|
|
|
|
- on_post_performance_check
|
|
|
|
|
|
|
|
|
|
Test loop
|
2020-02-27 12:35:20 +00:00
|
|
|
=========
|
2020-02-11 04:55:22 +00:00
|
|
|
|
|
|
|
|
- model.zero_grad()
|
|
|
|
|
- model.eval()
|
|
|
|
|
- torch.set_grad_enabled(False)
|
|
|
|
|
- test_step
|
|
|
|
|
- test_end
|
|
|
|
|
- model.train()
|
|
|
|
|
- torch.set_grad_enabled(True)
|
|
|
|
|
- on_post_performance_check
|
