update Loggers (#818)

* add warnings

* fix link
This commit is contained in:
Jirka Borovec
2020-02-21 13:39:37 -05:00
committed by GitHub
parent 9eb1907151
commit f7e9700aae
3 changed files with 5 additions and 3 deletions
+2 -1
View File
@@ -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
+2 -1
View File
@@ -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