mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-21 13:20:08 +08:00
19 lines
437 B
Python
19 lines
437 B
Python
from os import environ
|
|
from .base import LightningLoggerBase, rank_zero_only
|
|
|
|
try:
|
|
from .test_tube import TestTubeLogger
|
|
except ImportError:
|
|
pass
|
|
try:
|
|
from .mlflow import MLFlowLogger
|
|
except ImportError:
|
|
pass
|
|
try:
|
|
# needed to prevent ImportError and duplicated logs.
|
|
environ["COMET_DISABLE_AUTO_LOGGING"] = "1"
|
|
|
|
from .comet import CometLogger
|
|
except ImportError:
|
|
del environ["COMET_DISABLE_AUTO_LOGGING"]
|