From 3e5c5f9588700fd0c6b2ef8f6507913978eb2eb2 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 07:25:16 -0500 Subject: [PATCH] added direct links to docs --- pytorch_lightning/trainer/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pytorch_lightning/trainer/__init__.py b/pytorch_lightning/trainer/__init__.py index 30458845..9ce668ea 100644 --- a/pytorch_lightning/trainer/__init__.py +++ b/pytorch_lightning/trainer/__init__.py @@ -18,3 +18,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']