272 lines
8.8 KiB
ReStructuredText
272 lines
8.8 KiB
ReStructuredText
Experiment Logging
|
|
==================
|
|
|
|
Comet.ml
|
|
^^^^^^^^
|
|
|
|
`Comet.ml <https://www.comet.ml/site/>`_ is a third-party logger.
|
|
To use :class:`~pytorch_lightning.loggers.CometLogger` as your logger do the following.
|
|
First, install the package:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install comet-ml
|
|
|
|
Then configure the logger and pass it to the :class:`~pytorch_lightning.trainer.trainer.Trainer`:
|
|
|
|
.. doctest::
|
|
|
|
>>> import os
|
|
>>> from pytorch_lightning import Trainer
|
|
>>> from pytorch_lightning.loggers import CometLogger
|
|
>>> comet_logger = CometLogger(
|
|
... api_key=os.environ.get('COMET_API_KEY'),
|
|
... workspace=os.environ.get('COMET_WORKSPACE'), # Optional
|
|
... save_dir='.', # Optional
|
|
... project_name='default_project', # Optional
|
|
... rest_api_key=os.environ.get('COMET_REST_API_KEY'), # Optional
|
|
... experiment_name='default' # Optional
|
|
... )
|
|
>>> trainer = Trainer(logger=comet_logger)
|
|
|
|
The :class:`~pytorch_lightning.loggers.CometLogger` is available anywhere except ``__init__`` in your
|
|
:class:`~pytorch_lightning.core.lightning.LightningModule`.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import LightningModule
|
|
>>> class MyModule(LightningModule):
|
|
... def any_lightning_module_function_or_hook(self):
|
|
... some_img = fake_image()
|
|
... self.logger.experiment.add_image('generated_images', some_img, 0)
|
|
|
|
.. seealso::
|
|
:class:`~pytorch_lightning.loggers.CometLogger` docs.
|
|
|
|
MLflow
|
|
^^^^^^
|
|
|
|
`MLflow <https://mlflow.org/>`_ is a third-party logger.
|
|
To use :class:`~pytorch_lightning.loggers.MLFlowLogger` as your logger do the following.
|
|
First, install the package:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install mlflow
|
|
|
|
Then configure the logger and pass it to the :class:`~pytorch_lightning.trainer.trainer.Trainer`:
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import Trainer
|
|
>>> from pytorch_lightning.loggers import MLFlowLogger
|
|
>>> mlf_logger = MLFlowLogger(
|
|
... experiment_name="default",
|
|
... tracking_uri="file:/."
|
|
... )
|
|
>>> trainer = Trainer(logger=mlf_logger)
|
|
|
|
.. seealso::
|
|
:class:`~pytorch_lightning.loggers.MLFlowLogger` docs.
|
|
|
|
Neptune.ai
|
|
^^^^^^^^^^
|
|
|
|
`Neptune.ai <https://neptune.ai/>`_ is a third-party logger.
|
|
To use :class:`~pytorch_lightning.loggers.NeptuneLogger` as your logger do the following.
|
|
First, install the package:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install neptune-client
|
|
|
|
Then configure the logger and pass it to the :class:`~pytorch_lightning.trainer.trainer.Trainer`:
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import Trainer
|
|
>>> from pytorch_lightning.loggers import NeptuneLogger
|
|
>>> neptune_logger = NeptuneLogger(
|
|
... api_key='ANONYMOUS', # replace with your own
|
|
... project_name='shared/pytorch-lightning-integration',
|
|
... experiment_name='default', # Optional,
|
|
... params={'max_epochs': 10}, # Optional,
|
|
... tags=['pytorch-lightning', 'mlp'], # Optional,
|
|
... )
|
|
>>> trainer = Trainer(logger=neptune_logger)
|
|
|
|
The :class:`~pytorch_lightning.loggers.NeptuneLogger` is available anywhere except ``__init__`` in your
|
|
:class:`~pytorch_lightning.core.lightning.LightningModule`.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import LightningModule
|
|
>>> class MyModule(LightningModule):
|
|
... def any_lightning_module_function_or_hook(self):
|
|
... some_img = fake_image()
|
|
... self.logger.experiment.add_image('generated_images', some_img, 0)
|
|
|
|
.. seealso::
|
|
:class:`~pytorch_lightning.loggers.NeptuneLogger` docs.
|
|
|
|
allegro.ai TRAINS
|
|
^^^^^^^^^^^^^^^^^
|
|
|
|
`allegro.ai <https://github.com/allegroai/trains/>`_ is a third-party logger.
|
|
To use :class:`~pytorch_lightning.loggers.TrainsLogger` as your logger do the following.
|
|
First, install the package:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install trains
|
|
|
|
Then configure the logger and pass it to the :class:`~pytorch_lightning.trainer.trainer.Trainer`:
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import Trainer
|
|
>>> from pytorch_lightning.loggers import TrainsLogger
|
|
>>> trains_logger = TrainsLogger(
|
|
... project_name='examples',
|
|
... task_name='pytorch lightning test',
|
|
... ) # doctest: +ELLIPSIS
|
|
TRAINS Task: ...
|
|
TRAINS results page: ...
|
|
>>> trainer = Trainer(logger=trains_logger)
|
|
|
|
The :class:`~pytorch_lightning.loggers.TrainsLogger` is available anywhere in your
|
|
:class:`~pytorch_lightning.core.lightning.LightningModule`.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import LightningModule
|
|
>>> class MyModule(LightningModule):
|
|
... def __init__(self):
|
|
... some_img = fake_image()
|
|
... self.logger.experiment.log_image('debug', 'generated_image_0', some_img, 0)
|
|
|
|
.. seealso::
|
|
:class:`~pytorch_lightning.loggers.TrainsLogger` docs.
|
|
|
|
Tensorboard
|
|
^^^^^^^^^^^
|
|
|
|
To use `TensorBoard <https://pytorch.org/docs/stable/tensorboard.html>`_ as your logger do the following.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import Trainer
|
|
>>> from pytorch_lightning.loggers import TensorBoardLogger
|
|
>>> logger = TensorBoardLogger('tb_logs', name='my_model')
|
|
>>> trainer = Trainer(logger=logger)
|
|
|
|
The :class:`~pytorch_lightning.loggers.TensorBoardLogger` is available anywhere except ``__init__`` in your
|
|
:class:`~pytorch_lightning.core.lightning.LightningModule`.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import LightningModule
|
|
>>> class MyModule(LightningModule):
|
|
... def any_lightning_module_function_or_hook(self):
|
|
... some_img = fake_image()
|
|
... self.logger.experiment.add_image('generated_images', some_img, 0)
|
|
|
|
.. seealso::
|
|
:class:`~pytorch_lightning.loggers.TensorBoardLogger` docs.
|
|
|
|
Test Tube
|
|
^^^^^^^^^
|
|
|
|
`Test Tube <https://github.com/williamFalcon/test-tube>`_ is a
|
|
`TensorBoard <https://pytorch.org/docs/stable/tensorboard.html>`_ logger but with nicer file structure.
|
|
To use :class:`~pytorch_lightning.loggers.TestTubeLogger` as your logger do the following.
|
|
First, install the package:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install test_tube
|
|
|
|
Then configure the logger and pass it to the :class:`~pytorch_lightning.trainer.trainer.Trainer`:
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning.loggers import TestTubeLogger
|
|
>>> logger = TestTubeLogger('tb_logs', name='my_model')
|
|
>>> trainer = Trainer(logger=logger)
|
|
|
|
The :class:`~pytorch_lightning.loggers.TestTubeLogger` is available anywhere except ``__init__`` in your
|
|
:class:`~pytorch_lightning.core.lightning.LightningModule`.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import LightningModule
|
|
>>> class MyModule(LightningModule):
|
|
... def any_lightning_module_function_or_hook(self):
|
|
... some_img = fake_image()
|
|
... self.logger.experiment.add_image('generated_images', some_img, 0)
|
|
|
|
.. seealso::
|
|
:class:`~pytorch_lightning.loggers.TestTubeLogger` docs.
|
|
|
|
Weights and Biases
|
|
^^^^^^^^^^^^^^^^^^
|
|
|
|
`Weights and Biases <https://www.wandb.com/>`_ is a third-party logger.
|
|
To use :class:`~pytorch_lightning.loggers.WandbLogger` as your logger do the following.
|
|
First, install the package:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install wandb
|
|
|
|
Then configure the logger and pass it to the :class:`~pytorch_lightning.trainer.trainer.Trainer`:
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning.loggers import WandbLogger
|
|
>>> wandb_logger = WandbLogger()
|
|
>>> trainer = Trainer(logger=wandb_logger)
|
|
|
|
The :class:`~pytorch_lightning.loggers.WandbLogger` is available anywhere except ``__init__`` in your
|
|
:class:`~pytorch_lightning.core.lightning.LightningModule`.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import LightningModule
|
|
>>> class MyModule(LightningModule):
|
|
... def any_lightning_module_function_or_hook(self):
|
|
... some_img = fake_image()
|
|
... self.logger.experiment.log({
|
|
... "generated_images": [wandb.Image(some_img, caption="...")]
|
|
... })
|
|
|
|
.. seealso::
|
|
:class:`~pytorch_lightning.loggers.WandbLogger` docs.
|
|
|
|
Multiple Loggers
|
|
^^^^^^^^^^^^^^^^
|
|
|
|
Lightning supports the use of multiple loggers, just pass a list to the
|
|
:class:`~pytorch_lightning.trainer.trainer.Trainer`.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning.loggers import TensorBoardLogger, TestTubeLogger
|
|
>>> logger1 = TensorBoardLogger('tb_logs', name='my_model')
|
|
>>> logger2 = TestTubeLogger('tb_logs', name='my_model')
|
|
>>> trainer = Trainer(logger=[logger1, logger2])
|
|
|
|
The loggers are available as a list anywhere except ``__init__`` in your
|
|
:class:`~pytorch_lightning.core.lightning.LightningModule`.
|
|
|
|
.. doctest::
|
|
|
|
>>> from pytorch_lightning import LightningModule
|
|
>>> class MyModule(LightningModule):
|
|
... def any_lightning_module_function_or_hook(self):
|
|
... some_img = fake_image()
|
|
... # Option 1
|
|
... self.logger.experiment[0].add_image('generated_images', some_img, 0)
|
|
... # Option 2
|
|
... self.logger[0].experiment.add_image('generated_images', some_img, 0)
|