easy import for lightningModule

2019-07-03 18:43:13 -04:00 · 2019-07-03 18:43:13 -04:00 · ae81473464
parent 153b95c01f
commit ae81473464
1 changed files with 2 additions and 1 deletions
--- a/pytorch_lightning/init.py
+++ b/pytorch_lightning/init.py
@ -1 +1,2 @@
-from .models import Trainer
+from .models import Trainer
+from .root_module.root_module import LightningModule