mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-08-30 11:29:56 +08:00
140 lines
3.5 KiB
ReStructuredText
140 lines
3.5 KiB
ReStructuredText
Experiment Logging
|
|
===================
|
|
|
|
Comet.ml
|
|
^^^^^^^^
|
|
|
|
`Comet.ml <https://www.comet.ml/site/>`_ 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 <https://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 <https://pytorch.org/docs/stable/tensorboard.html>`_ 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 <https://github.com/williamFalcon/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 <https://www.wandb.com/>`_ 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)
|
|
|