change default logger to dedicated one

Author: cmpute

pytorch_lightning/callbacks/early_stopping.py

@@ -5,13 +5,12 @@
 
 """
 
-import logging as log
 import warnings
 
 import numpy as np
 
 from .base import Callback
-
+from ..core import root_logger
 
 class EarlyStopping(Callback):
     r"""

pytorch_lightning/callbacks/model_checkpoint.py

@@ -7,13 +7,13 @@
 
 import os
 import shutil
-import logging as log
 import warnings
 import re
 
 import numpy as np
 
 from pytorch_lightning.callbacks.base import Callback
+from pytorch_lightning.core import root_logger as log
 
 
 class ModelCheckpoint(Callback):

pytorch_lightning/core/__init__.py

@@ -312,6 +312,9 @@ def test_dataloader(self):
 
 """
 
+import logging
+root_logger = logging.getLogger("lightning")
+
 from .decorators import data_loader
 from .lightning import LightningModule
 

pytorch_lightning/core/lightning.py

@@ -1,6 +1,5 @@
 import collections
 import inspect
-import logging as log
 import os
 import warnings
 from abc import ABC, abstractmethod
@@ -11,6 +10,7 @@
 import torch.distributed as dist
 from torch.optim import Adam
 
+from pytorch_lightning.core import root_logger as log
 from pytorch_lightning.core.grads import GradInformation
 from pytorch_lightning.core.hooks import ModelHooks
 from pytorch_lightning.core.saving import ModelIO, load_hparams_from_tags_csv
@@ -25,7 +25,6 @@
 except ImportError:
     XLA_AVAILABLE = False
 
-
 class LightningModule(ABC, GradInformation, ModelIO, ModelHooks):
 
     def __init__(self, *args, **kwargs):

pytorch_lightning/core/memory.py

@@ -3,14 +3,14 @@
 '''
 
 import gc
-import logging as log
 import os
 import subprocess
 from subprocess import PIPE
 
 import numpy as np
 import torch
 
+from ..core import root_logger as log
 
 class ModelSummary(object):
 

pytorch_lightning/core/saving.py

@@ -1,8 +1,9 @@
-import os
 import csv
-import logging as log
+import os
 from argparse import Namespace
 
+from ..core import root_logger as log
+
 
 class ModelIO(object):
 

pytorch_lightning/profiler/profiler.py

@@ -6,7 +6,7 @@
 import pstats
 import io
 from abc import ABC, abstractmethod
-import logging as log
+from ..core import root_logger as log
 
 
 class BaseProfiler(ABC):

pytorch_lightning/trainer/auto_mix_precision.py

@@ -1,13 +1,14 @@
 
 from abc import ABC
 
+from ..core import root_logger as log
+
 try:
     from apex import amp
 
     APEX_AVAILABLE = True
 except ImportError:
     APEX_AVAILABLE = False
-import logging as log
 
 
 class TrainerAMPMixin(ABC):

pytorch_lightning/trainer/distrib_data_parallel.py

@@ -113,14 +113,14 @@ def train_fx(trial_hparams, cluster_manager, _):
 
 """
 
-import logging as log
 import os
 import re
 import warnings
 from abc import ABC, abstractmethod
 from typing import Union
 
 import torch
+from pytorch_lightning.core import root_logger as log
 from pytorch_lightning.loggers import LightningLoggerBase
 
 from pytorch_lightning.utilities.debugging import MisconfigurationException

pytorch_lightning/trainer/distrib_parts.py

@@ -335,12 +335,12 @@
 """
 
 from abc import ABC, abstractmethod
-import logging as log
 import os
 import signal
 
 import torch
 
+from pytorch_lightning.core import root_logger as log
 from pytorch_lightning.overrides.data_parallel import (
     LightningDistributedDataParallel,
     LightningDataParallel,

pytorch_lightning/trainer/trainer.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import warnings
-import logging as log
 from typing import Union, Optional, List, Dict, Tuple, Iterable
 from argparse import ArgumentParser
 
@@ -13,6 +12,7 @@
 from tqdm.auto import tqdm
 from torch.optim.optimizer import Optimizer
 
+from pytorch_lightning.core import root_logger as log
 from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
 from pytorch_lightning.loggers import LightningLoggerBase
 from pytorch_lightning.profiler.profiler import BaseProfiler

pytorch_lightning/trainer/training_io.py

@@ -102,6 +102,7 @@
 import torch
 import torch.distributed as dist
 
+from pytorch_lightning.core import root_logger as log
 from pytorch_lightning.core.lightning import LightningModule
 from pytorch_lightning.loggers import LightningLoggerBase
 from pytorch_lightning.overrides.data_parallel import (

pytorch_lightning/trainer/training_loop.py

@@ -125,13 +125,13 @@ def training_step(self, batch, batch_idx):
 
 import copy
 import warnings
-import logging as log
 from abc import ABC, abstractmethod
 from typing import Union, List
 
 import numpy as np
 from torch.utils.data import DataLoader
 
+from pytorch_lightning.core import root_logger as log
 from pytorch_lightning.core.lightning import LightningModule
 from pytorch_lightning.loggers import LightningLoggerBase
 from pytorch_lightning.utilities.debugging import MisconfigurationException

pytorch_lightning/trainer/training_tricks.py

@@ -1,9 +1,9 @@
-import logging as log
 from abc import ABC, abstractmethod
 
 import torch
 import math
 
+from pytorch_lightning.core import root_logger as log
 from pytorch_lightning.callbacks import GradientAccumulationScheduler
 
 EPSILON = 1e-6