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)