mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-08-25 11:24:04 +08:00
Jirka Borovec
d0ccbb8e1c
* fix loading package while setup #56 * Update setup.py * Update __init__.py * Update setup.py * use complete req. * Update setup.py * Update setup.py
10 lines
209 B
Python
10 lines
209 B
Python
from .models.trainer import Trainer
|
|
from .root_module.root_module import LightningModule
|
|
from .root_module.decorators import data_loader
|
|
|
|
__all__ = [
|
|
'Trainer',
|
|
'LightningModule',
|
|
'data_loader',
|
|
]
|