From c3d3c47554b8c2cdd725ca45a388b5489b3e4158 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 17:36:05 -0500 Subject: [PATCH] set auto dp if no backend --- pytorch_lightning/logging/wandb.py | 34 +++++++++--------------------- 1 file changed, 10 insertions(+), 24 deletions(-) diff --git a/pytorch_lightning/logging/wandb.py b/pytorch_lightning/logging/wandb.py index d076aad2..829006d8 100644 --- a/pytorch_lightning/logging/wandb.py +++ b/pytorch_lightning/logging/wandb.py @@ -1,27 +1,3 @@ -""" -Log using `W&B `_ - -.. code-block:: python - - >>> from pytorch_lightning.logging import WandbLogger - >>> from pytorch_lightning import Trainer - >>> wandb_logger = WandbLogger() - >>> trainer = Trainer(logger=wandb_logger) - - -Use the logger anywhere in you LightningModule as follows: - -.. code-block:: python - - def train_step(...): - # example - self.logger.experiment.whatever_wandb_supports(...) - - def any_lightning_module_function_or_hook(...): - self.logger.experiment.whatever_wandb_supports(...) - -""" - import os try: @@ -44,6 +20,16 @@ class WandbLogger(LightningLoggerBase): anonymous (bool): enables or explicitly disables anonymous logging. project (str): the name of the project to which this run will belong. tags (list of str): tags associated with this run. + + Example + -------- + .. code-block:: python + + from pytorch_lightning.logging import WandbLogger + from pytorch_lightning import Trainer + + wandb_logger = WandbLogger() + trainer = Trainer(logger=wandb_logger) """ def __init__(self, name=None, save_dir=None, offline=False, id=None, anonymous=False,