From f7e9700aae07bac5ab153534330ff172da59b617 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Fri, 21 Feb 2020 19:39:37 +0100 Subject: [PATCH] update Loggers (#818) * add warnings * fix link --- README.md | 2 +- pytorch_lightning/loggers/neptune.py | 3 ++- pytorch_lightning/loggers/wandb.py | 3 ++- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index dd83c765..fa8c1f1d 100644 --- a/README.md +++ b/README.md @@ -295,7 +295,7 @@ Lightning also adds a text column with all the hyperparameters for this experime - [Running grid search on a cluster](https://pytorch-lightning.rtfd.io/en/latest/pytorch_lightning.trainer.distrib_data_parallel.html) - [Fast dev run](https://pytorch-lightning.rtfd.io/en/latest/pytorch_lightning.utilities.debugging.html) -- [Logging](https://pytorch-lightning.rtfd.io/en/latest/pytorch_lightning.logging.html) +- [Logging](https://pytorch-lightning.rtfd.io/en/latest/pytorch_lightning.loggers.html) - [Implement Your Own Distributed (DDP) training](https://pytorch-lightning.rtfd.io/en/latest/pytorch_lightning.core.lightning.html#pytorch_lightning.core.lightning.LightningModule.configure_ddp) - [Multi-GPU & Multi-node](https://pytorch-lightning.rtfd.io/en/latest/pytorch_lightning.trainer.distrib_parts.html) - [Training loop](https://pytorch-lightning.rtfd.io/en/latest/pytorch_lightning.trainer.training_loop.html) diff --git a/pytorch_lightning/loggers/neptune.py b/pytorch_lightning/loggers/neptune.py index e8a087a0..f81730b6 100644 --- a/pytorch_lightning/loggers/neptune.py +++ b/pytorch_lightning/loggers/neptune.py @@ -12,7 +12,8 @@ from logging import getLogger try: import neptune except ImportError: - raise ImportError('Missing neptune package. Run `pip install neptune-client`') + raise ImportError('You want to use `neptune` logger which is not installed yet,' + ' please install it e.g. `pip install neptune-client`.') from torch import is_tensor diff --git a/pytorch_lightning/loggers/wandb.py b/pytorch_lightning/loggers/wandb.py index c55f1e6c..f3ddede7 100644 --- a/pytorch_lightning/loggers/wandb.py +++ b/pytorch_lightning/loggers/wandb.py @@ -11,7 +11,8 @@ import os try: import wandb except ImportError: - raise ImportError('Missing wandb package.') + raise ImportError('You want to use `wandb` logger which is not installed yet,' + ' please install it e.g. `pip install wandb`.') from .base import LightningLoggerBase, rank_zero_only