lightning/pytorch_lightning/__init__.py

14 lines
445 B
Python

from .models.trainer import Trainer
from .root_module.root_module import LightningModule
from .root_module.decorators import data_loader
__version__ = '0.3.6.9'
__author__ = "William Falcon",
__author_email__ = "waf2107@columbia.edu"
__license__ = 'MIT'
__homepage__ = 'https://github.com/williamFalcon/pytorch-lightning',
__copyright__ = 'Copyright (c) 2018-2019, %s.' % __author__
__doc__ = """
The Keras for ML researchers using PyTorch
"""