20 lines
517 B
Python
20 lines
517 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
|
|
"""
|
|
|
|
__all__ = [
|
|
'Trainer',
|
|
'LightningModule',
|
|
'data_loader',
|
|
]
|