Experiment Logging
===================
Comet.ml
^^^^^^^^
`Comet.ml `_ is a third-party logger.
To use CometLogger as your logger do the following.
.. note:: See: :ref:`comet` docs.
.. code-block:: python
from pytorch_lightning.loggers import TestTubeLogger
comet_logger = CometLogger(
api_key=os.environ["COMET_KEY"],
workspace=os.environ["COMET_WORKSPACE"], # Optional
project_name="default_project", # Optional
rest_api_key=os.environ["COMET_REST_KEY"], # Optional
experiment_name="default" # Optional
)
trainer = Trainer(logger=comet_logger)
The CometLogger is available anywhere in your LightningModule
.. code-block:: python
class MyModule(pl.LightningModule):
def __init__(self, ...):
some_img = fake_image()
self.logger.experiment.add_image('generated_images', some_img, 0)
Neptune.ai
^^^^^^^^^^
`Neptune.ai `_ is a third-party logger.
To use Neptune.ai as your logger do the following.
.. note:: See: :ref:`neptune` docs.
.. code-block:: python
from pytorch_lightning.loggers import NeptuneLogger
neptune_logger = NeptuneLogger(
project_name="USER_NAME/PROJECT_NAME",
experiment_name="default", # Optional,
params={"max_epochs": 10}, # Optional,
tags=["pytorch-lightning","mlp"] # Optional,
)
trainer = Trainer(logger=neptune_logger)
The Neptune.ai is available anywhere in your LightningModule
.. code-block:: python
class MyModule(pl.LightningModule):
def __init__(self, ...):
some_img = fake_image()
self.logger.experiment.add_image('generated_images', some_img, 0)
Tensorboard
^^^^^^^^^^^
To use `Tensorboard `_ as your logger do the following.
.. note:: See: TensorBoardLogger :ref:`tf-logger`
.. code-block:: python
from pytorch_lightning.loggers import TensorBoardLogger
logger = TensorBoardLogger("tb_logs", name="my_model")
trainer = Trainer(logger=logger)
The TensorBoardLogger is available anywhere in your LightningModule
.. code-block:: python
class MyModule(pl.LightningModule):
def __init__(self, ...):
some_img = fake_image()
self.logger.experiment.add_image('generated_images', some_img, 0)
Test Tube
^^^^^^^^^
`Test Tube `_ is a tensorboard logger but with nicer file structure.
To use TestTube as your logger do the following.
.. note:: See: TestTube :ref:`testTube`
.. code-block:: python
from pytorch_lightning.loggers import TestTubeLogger
logger = TestTubeLogger("tb_logs", name="my_model")
trainer = Trainer(logger=logger)
The TestTubeLogger is available anywhere in your LightningModule
.. code-block:: python
class MyModule(pl.LightningModule):
def __init__(self, ...):
some_img = fake_image()
self.logger.experiment.add_image('generated_images', some_img, 0)
Wandb
^^^^^
`Wandb `_ is a third-party logger.
To use Wandb as your logger do the following.
.. note:: See: :ref:`wandb` docs
.. code-block:: python
from pytorch_lightning.loggers import WandbLogger
wandb_logger = WandbLogger()
trainer = Trainer(logger=wandb_logger)
The Wandb logger is available anywhere in your LightningModule
.. code-block:: python
class MyModule(pl.LightningModule):
def __init__(self, ...):
some_img = fake_image()
self.logger.experiment.add_image('generated_images', some_img, 0)