wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-28 13:20:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

easy import for lightningModule

Browse Source
This commit is contained in:
William Falcon committed 2019-07-03 18:43:13 -04:00
1 parent 153b95c01f
commit ae81473464
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/__init__.py
+2 -1
View File
@@ -1 +1,2 @@
from .models import Trainer
from .models import Trainer
from .root_module.root_module import LightningModule