diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py index ce4b32be27..5893eb4a97 100644 --- a/pytorch_lightning/__init__.py +++ b/pytorch_lightning/__init__.py @@ -1 +1,2 @@ -from .models import Trainer \ No newline at end of file +from .models import Trainer +from .root_module.root_module import LightningModule \ No newline at end of file