diff --git a/pytorch_lightning/trainer/__init__.py b/pytorch_lightning/trainer/__init__.py index 318ec82d..4993c20c 100644 --- a/pytorch_lightning/trainer/__init__.py +++ b/pytorch_lightning/trainer/__init__.py @@ -22,3 +22,7 @@ The LightningTemplate holds the core computations, train, val, test loop, optimi This pattern de-couples the engineering from the science which makes your code reusable and free to run on any hardware. """ + +from .trainer import Trainer + +__all__ = ['Trainer']