From 9f9bf65edeab5e4442446e216f0806f5f1dd1931 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 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']