diff --git a/pytorch_lightning/trainer/__init__.py b/pytorch_lightning/trainer/__init__.py index c18f22b8..e250350d 100644 --- a/pytorch_lightning/trainer/__init__.py +++ b/pytorch_lightning/trainer/__init__.py @@ -11,6 +11,7 @@ The trainer automates all parts of training except: - which optimizers to use - how to do the computations + The Trainer delegates those calls to your LightningModule which defines how to do those parts. This is the basic use of the trainer: