2020-02-01 23:24:56 +00:00
|
|
|
"""Root package info."""
|
2019-10-28 22:41:13 +00:00
|
|
|
|
2020-09-11 19:59:36 +00:00
|
|
|
__version__ = '0.9.1rc3'
|
2019-11-28 17:48:55 +00:00
|
|
|
__author__ = 'William Falcon et al.'
|
2019-10-28 22:41:13 +00:00
|
|
|
__author_email__ = 'waf2107@columbia.edu'
|
|
|
|
__license__ = 'Apache-2.0'
|
2020-02-01 23:24:56 +00:00
|
|
|
__copyright__ = 'Copyright (c) 2018-2020, %s.' % __author__
|
2020-01-20 19:50:31 +00:00
|
|
|
__homepage__ = 'https://github.com/PyTorchLightning/pytorch-lightning'
|
2019-11-09 05:59:14 +00:00
|
|
|
# this has to be simple string, see: https://github.com/pypa/twine/issues/522
|
2020-07-24 15:42:15 +00:00
|
|
|
__docs__ = (
|
|
|
|
"PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers."
|
|
|
|
" Scale your models. Write less boilerplate."
|
|
|
|
)
|
2020-04-20 21:36:26 +00:00
|
|
|
__long_docs__ = """
|
|
|
|
Lightning is a way to organize your PyTorch code to decouple the science code from the engineering.
|
|
|
|
It's more of a style-guide than a framework.
|
|
|
|
|
|
|
|
In Lightning, you organize your code into 3 distinct categories:
|
|
|
|
|
|
|
|
1. Research code (goes in the LightningModule).
|
|
|
|
2. Engineering code (you delete, and is handled by the Trainer).
|
|
|
|
3. Non-essential research code (logging, etc. this goes in Callbacks).
|
|
|
|
|
|
|
|
Although your research/production project might start simple, once you add things like GPU AND TPU training,
|
|
|
|
16-bit precision, etc, you end up spending more time engineering than researching.
|
|
|
|
Lightning automates AND rigorously tests those parts for you.
|
|
|
|
|
|
|
|
Overall, Lightning guarantees rigorously tested, correct, modern best practices for the automated parts.
|
|
|
|
|
|
|
|
Documentation
|
|
|
|
-------------
|
|
|
|
- https://pytorch-lightning.readthedocs.io/en/latest
|
|
|
|
- https://pytorch-lightning.readthedocs.io/en/stable
|
|
|
|
"""
|
2019-10-28 22:41:13 +00:00
|
|
|
|
2020-03-29 18:56:36 +00:00
|
|
|
import logging as python_logging
|
2020-03-20 19:49:01 +00:00
|
|
|
|
2020-03-29 18:56:36 +00:00
|
|
|
_logger = python_logging.getLogger("lightning")
|
2020-05-12 04:14:35 +00:00
|
|
|
_logger.addHandler(python_logging.StreamHandler())
|
|
|
|
_logger.setLevel(python_logging.INFO)
|
2020-03-20 19:49:01 +00:00
|
|
|
|
2019-10-28 22:41:13 +00:00
|
|
|
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__:
|
2020-03-19 13:14:29 +00:00
|
|
|
import sys # pragma: no-cover
|
2020-07-24 15:42:15 +00:00
|
|
|
|
2020-03-20 19:49:01 +00:00
|
|
|
sys.stdout.write(f'Partial import of `{__name__}` during the build process.\n') # pragma: no-cover
|
2020-03-19 13:14:29 +00:00
|
|
|
# We are not importing the rest of the lightning during the build process, as it may not be compiled yet
|
2019-10-28 22:41:13 +00:00
|
|
|
else:
|
2020-07-30 21:19:28 +00:00
|
|
|
from pytorch_lightning.core import LightningDataModule, LightningModule
|
2020-07-27 23:07:09 +00:00
|
|
|
from pytorch_lightning.core.step_result import TrainResult, EvalResult
|
2020-06-18 11:21:44 +00:00
|
|
|
from pytorch_lightning.callbacks import Callback
|
2020-03-19 13:14:29 +00:00
|
|
|
from pytorch_lightning.trainer import Trainer
|
2020-06-17 17:42:28 +00:00
|
|
|
from pytorch_lightning.utilities.seed import seed_everything
|
2020-07-05 03:55:22 +00:00
|
|
|
from pytorch_lightning import metrics
|
2019-10-28 22:41:13 +00:00
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
'Trainer',
|
2020-07-24 15:42:15 +00:00
|
|
|
'LightningDataModule',
|
2019-10-28 22:41:13 +00:00
|
|
|
'LightningModule',
|
2020-02-26 04:17:27 +00:00
|
|
|
'Callback',
|
2020-06-03 01:45:19 +00:00
|
|
|
'seed_everything',
|
2020-07-20 23:00:20 +00:00
|
|
|
'metrics',
|
|
|
|
'EvalResult',
|
2020-07-24 15:42:15 +00:00
|
|
|
'TrainResult',
|
2019-10-28 22:41:13 +00:00
|
|
|
]
|
2020-04-21 11:12:02 +00:00
|
|
|
|
|
|
|
# necessary for regular bolts imports. Skip exception since bolts is not always installed
|
|
|
|
try:
|
|
|
|
from pytorch_lightning import bolts
|
|
|
|
except ImportError:
|
|
|
|
pass
|
2020-02-09 22:39:10 +00:00
|
|
|
# __call__ = __all__
|
2020-04-21 11:12:02 +00:00
|
|
|
|
|
|
|
# for compatibility with namespace packages
|
|
|
|
__import__('pkg_resources').declare_namespace(__name__)
|