mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-21 13:20:08 +08:00
Jirka Borovec
d71556e7a1
* upgrade req. * move MkDocs * create Sphinx * init Sphinx * move md from MkDocs to Sphinx * CI: build docs * build Sphinx formatting move docs from MD to docstring in particular package/modules formatting add Sphinx ext. rename root_module to core drop implicit name "_logger" drop duplicate name "overwrite" fix imports use pytorch theme add sample link mapping try fix RTD build use forked template fix some docs warnings fix paths add deprecation warnings fix flake8 fix paths revert refactor revert MLFlowLogger * revert example import * update link * Update lightning_module_template.py
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""Package info"""
|
|
|
|
__version__ = '0.5.3.2'
|
|
__author__ = 'William Falcon et al.'
|
|
__author_email__ = 'waf2107@columbia.edu'
|
|
__license__ = 'Apache-2.0'
|
|
__copyright__ = 'Copyright (c) 2018-2019, %s.' % __author__
|
|
__homepage__ = 'https://github.com/williamFalcon/pytorch-lightning'
|
|
# this has to be simple string, see: https://github.com/pypa/twine/issues/522
|
|
__docs__ = "PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers." \
|
|
" Scale your models. Write less boilerplate."
|
|
|
|
|
|
try:
|
|
# This variable is injected in the __builtins__ by the build
|
|
# process. It used to enable importing subpackages of skimage when
|
|
# the binaries are not built
|
|
__LIGHTNING_SETUP__
|
|
except NameError:
|
|
__LIGHTNING_SETUP__ = False
|
|
|
|
if __LIGHTNING_SETUP__:
|
|
import sys
|
|
sys.stderr.write('Partial import of skimage during the build process.\n')
|
|
# We are not importing the rest of the scikit during the build
|
|
# process, as it may not be compiled yet
|
|
else:
|
|
from .trainer.trainer import Trainer
|
|
from .core.lightning import LightningModule
|
|
from .core.decorators import data_loader
|
|
|
|
__all__ = [
|
|
'Trainer',
|
|
'LightningModule',
|
|
'data_loader',
|
|
]
|