lightning/pytorch_lightning/trainer/__init__.py

20 lines
457 B
Python
Raw Normal View History

"""
# Trainer
The lightning trainer abstracts best practices for running a training, val, test routine.
It calls parts of your model when it wants to hand over full control and otherwise makes
training assumptions which are now standard practice in AI research.
This is the basic use of the trainer:
.. code-block:: python
from pytorch_lightning import Trainer
model = LightningTemplate()
trainer = Trainer()
trainer.fit(model)
"""