lightning/pytorch_lightning/__init__.py

3 lines
136 B
Python
Raw Normal View History

2019-08-03 15:08:51 +00:00
from .models.trainer import Trainer
from .root_module.root_module import LightningModule
from .root_module.decorators import data_loader