mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-01 12:32:04 +08:00
William Falcon
7099f8dbfb
* split trainer mixins * Update multi_node_cluster_template.py * Update single_cpu_template.py * Update single_gpu_node_16bit_template.py * Update single_gpu_node_ddp_template.py * Update single_gpu_node_dp_template.py * Update trainer_cpu_template.py * Update trainer_io.py * split trainer mixins * Update multi_node_cluster_template.py * deconflicted * deconflicted * deconflicted
10 lines
210 B
Python
10 lines
210 B
Python
from .trainer.trainer import Trainer
|
|
from .root_module.root_module import LightningModule
|
|
from .root_module.decorators import data_loader
|
|
|
|
__all__ = [
|
|
'Trainer',
|
|
'LightningModule',
|
|
'data_loader',
|
|
]
|