2019-08-03 15:08:51 +00:00
|
|
|
from .models.trainer import Trainer
|
2019-07-25 14:56:03 +00:00
|
|
|
from .root_module.root_module import LightningModule
|
2019-08-05 20:19:55 +00:00
|
|
|
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
|
|
|
|
|
"""
|
2019-08-05 21:57:39 +00:00
|
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
|
'Trainer',
|
|
|
|
|
'LightningModule',
|
|
|
|
|
'data_loader',
|
|
|
|
|
]
|
