name

Author: Borda

pytorch_lightning/loggers/wandb.py

@@ -24,7 +24,7 @@
 
 from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_experiment
 from pytorch_lightning.utilities import rank_zero_only, _module_available
-from pytorch_lightning.utilities.warning_utils import WarningCache
+from pytorch_lightning.utilities.warning import WarningCache
 
 _WANDB_AVAILABLE = _module_available("wandb")
 

pytorch_lightning/overrides/data_parallel.py

@@ -24,7 +24,7 @@
 from torch.nn.parallel._functions import Gather
 
 from pytorch_lightning.core.step_result import Result
-from pytorch_lightning.utilities.warning_utils import WarningCache
+from pytorch_lightning.utilities.warning import WarningCache
 
 
 def _find_tensors(obj):  # pragma: no-cover

pytorch_lightning/trainer/evaluation_loop.py

@@ -18,7 +18,7 @@
 from pytorch_lightning.utilities.distributed import rank_zero_warn
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from pytorch_lightning.utilities.model_utils import is_overridden
-from pytorch_lightning.utilities.warning_utils import WarningCache
+from pytorch_lightning.utilities.warning import WarningCache
 
 
 class EvaluationLoop(object):

pytorch_lightning/trainer/training_loop.py

@@ -32,7 +32,7 @@
 from pytorch_lightning.utilities.memory import recursive_detach
 from pytorch_lightning.utilities.model_utils import is_overridden
 from pytorch_lightning.utilities.parsing import AttributeDict
-from pytorch_lightning.utilities.warning_utils import WarningCache
+from pytorch_lightning.utilities.warning import WarningCache
 
 
 class TrainLoop: