Skip to content

Commit 959abc2

Browse files
committed
migrate ignite.contrib.handlers
move loggers and other class from contrib to ignite.handlers make changes in a backwards compatible manner fix circular imports that happened during the migration in ignite/__init__.py and ignite/handlers/__init__.py
1 parent 3d8f3ec commit 959abc2

26 files changed

+4846
-4523
lines changed

docs/source/contrib/handlers.rst

Lines changed: 2 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -28,37 +28,5 @@ Time profilers [deprecated]
2828
Loggers
2929
-------
3030

31-
.. currentmodule:: ignite.contrib.handlers
32-
33-
.. autosummary::
34-
:nosignatures:
35-
:toctree: ../generated
36-
:recursive:
37-
38-
base_logger
39-
clearml_logger
40-
mlflow_logger
41-
neptune_logger
42-
polyaxon_logger
43-
tensorboard_logger
44-
tqdm_logger
45-
46-
visdom_logger
47-
wandb_logger
48-
49-
.. seealso::
50-
51-
Below are a comprehensive list of examples of various loggers.
52-
53-
* See `tensorboardX mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_tensorboard_logger.py>`_
54-
and `CycleGAN and EfficientNet notebooks <https://github.com/pytorch/ignite/tree/master/examples/notebooks>`_ for detailed usage.
55-
56-
* See `visdom mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_visdom_logger.py>`_ for detailed usage.
57-
58-
* See `neptune mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_neptune_logger.py>`_ for detailed usage.
59-
60-
* See `tqdm mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_tqdm_logger.py>`_ for detailed usage.
61-
62-
* See `wandb mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_wandb_logger.py>`_ for detailed usage.
63-
64-
* See `clearml mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_clearml_logger.py>`_ for detailed usage.
31+
.. deprecated:: 0.4.14
32+
Loggers moved to ignite.handlers.

docs/source/handlers.rst

Lines changed: 43 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
ignite.handlers
22
===============
33

4-
Complete list of handlers
5-
-------------------------
4+
Complete list of generic handlers
5+
----------------------------------
66

77
.. currentmodule:: ignite.handlers
88

@@ -33,6 +33,46 @@ Complete list of handlers
3333
param_scheduler.ParamScheduler
3434
state_param_scheduler.StateParamScheduler
3535

36+
37+
Loggers
38+
--------
39+
40+
.. currentmodule:: ignite.handlers
41+
42+
.. autosummary::
43+
:nosignatures:
44+
:toctree: generated
45+
:recursive:
46+
47+
base_logger
48+
clearml_logger
49+
mlflow_logger
50+
neptune_logger
51+
polyaxon_logger
52+
tensorboard_logger
53+
tqdm_logger
54+
55+
visdom_logger
56+
wandb_logger
57+
58+
.. seealso::
59+
60+
Below are a comprehensive list of examples of various loggers.
61+
62+
* See `tensorboardX mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_tensorboard_logger.py>`_
63+
and `CycleGAN and EfficientNet notebooks <https://github.com/pytorch/ignite/tree/master/examples/notebooks>`_ for detailed usage.
64+
65+
* See `visdom mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_visdom_logger.py>`_ for detailed usage.
66+
67+
* See `neptune mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_neptune_logger.py>`_ for detailed usage.
68+
69+
* See `tqdm mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_tqdm_logger.py>`_ for detailed usage.
70+
71+
* See `wandb mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_wandb_logger.py>`_ for detailed usage.
72+
73+
* See `clearml mnist example <https://github.com/pytorch/ignite/blob/master/examples/mnist/mnist_with_clearml_logger.py>`_ for detailed usage.
74+
75+
3676
.. _param-scheduler-label:
3777

3878
Parameter scheduler
@@ -396,7 +436,7 @@ Example with :class:`ignite.handlers.param_scheduler.ReduceLROnPlateauScheduler`
396436
init_lr = 0.1
397437
398438
lr_values = np.array(ReduceLROnPlateauScheduler.simulate_values(
399-
num_events, metric_values, init_lr,
439+
num_events, metric_values, init_lr,
400440
factor=0.5, patience=1, mode='max', threshold=0.01, threshold_mode='abs'
401441
)
402442
)

ignite/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1+
__version__ = "0.5.0"
2+
13
import ignite.contrib
24
import ignite.distributed
35
import ignite.engine
46
import ignite.exceptions
57
import ignite.handlers
68
import ignite.metrics
79
import ignite.utils
8-
9-
__version__ = "0.5.0"

ignite/contrib/engines/common.py

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -15,21 +15,24 @@
1515
from torch.optim.lr_scheduler import _LRScheduler as PyTorchLRScheduler
1616

1717
import ignite.distributed as idist
18-
from ignite.contrib.handlers import (
18+
from ignite.contrib.metrics import GpuInfo
19+
from ignite.engine import Engine, Events
20+
from ignite.handlers import (
21+
Checkpoint,
1922
ClearMLLogger,
23+
DiskSaver,
24+
EarlyStopping,
2025
global_step_from_engine,
2126
MLflowLogger,
2227
NeptuneLogger,
2328
PolyaxonLogger,
2429
ProgressBar,
2530
TensorboardLogger,
31+
TerminateOnNan,
2632
VisdomLogger,
2733
WandBLogger,
2834
)
29-
from ignite.contrib.handlers.base_logger import BaseLogger
30-
from ignite.contrib.metrics import GpuInfo
31-
from ignite.engine import Engine, Events
32-
from ignite.handlers import Checkpoint, DiskSaver, EarlyStopping, TerminateOnNan
35+
from ignite.handlers.base_logger import BaseLogger
3336
from ignite.handlers.checkpoint import BaseSaveHandler
3437
from ignite.handlers.param_scheduler import ParamScheduler
3538
from ignite.metrics import RunningAverage
@@ -361,7 +364,7 @@ def setup_tb_logging(
361364
kwargs: optional keyword args to be passed to construct the logger.
362365
363366
Returns:
364-
:class:`~ignite.contrib.handlers.tensorboard_logger.TensorboardLogger`
367+
:class:`~ignite.handlers.tensorboard_logger.TensorboardLogger`
365368
"""
366369
logger = TensorboardLogger(log_dir=output_path, **kwargs)
367370
_setup_logging(logger, trainer, optimizers, evaluators, log_every_iters)
@@ -392,7 +395,7 @@ def setup_visdom_logging(
392395
kwargs: optional keyword args to be passed to construct the logger.
393396
394397
Returns:
395-
:class:`~ignite.contrib.handlers.visdom_logger.VisdomLogger`
398+
:class:`~ignite.handlers.visdom_logger.VisdomLogger`
396399
"""
397400
logger = VisdomLogger(**kwargs)
398401
_setup_logging(logger, trainer, optimizers, evaluators, log_every_iters)
@@ -423,7 +426,7 @@ def setup_mlflow_logging(
423426
kwargs: optional keyword args to be passed to construct the logger.
424427
425428
Returns:
426-
:class:`~ignite.contrib.handlers.mlflow_logger.MLflowLogger`
429+
:class:`~ignite.handlers.mlflow_logger.MLflowLogger`
427430
"""
428431
logger = MLflowLogger(**kwargs)
429432
_setup_logging(logger, trainer, optimizers, evaluators, log_every_iters)
@@ -454,7 +457,7 @@ def setup_neptune_logging(
454457
kwargs: optional keyword args to be passed to construct the logger.
455458
456459
Returns:
457-
:class:`~ignite.contrib.handlers.neptune_logger.NeptuneLogger`
460+
:class:`~ignite.handlers.neptune_logger.NeptuneLogger`
458461
"""
459462
logger = NeptuneLogger(**kwargs)
460463
_setup_logging(logger, trainer, optimizers, evaluators, log_every_iters)
@@ -485,7 +488,7 @@ def setup_wandb_logging(
485488
kwargs: optional keyword args to be passed to construct the logger.
486489
487490
Returns:
488-
:class:`~ignite.contrib.handlers.wandb_logger.WandBLogger`
491+
:class:`~ignite.handlers.wandb_logger.WandBLogger`
489492
"""
490493
logger = WandBLogger(**kwargs)
491494
_setup_logging(logger, trainer, optimizers, evaluators, log_every_iters)
@@ -516,7 +519,7 @@ def setup_plx_logging(
516519
kwargs: optional keyword args to be passed to construct the logger.
517520
518521
Returns:
519-
:class:`~ignite.contrib.handlers.polyaxon_logger.PolyaxonLogger`
522+
:class:`~ignite.handlers.polyaxon_logger.PolyaxonLogger`
520523
"""
521524
logger = PolyaxonLogger(**kwargs)
522525
_setup_logging(logger, trainer, optimizers, evaluators, log_every_iters)
@@ -547,7 +550,7 @@ def setup_clearml_logging(
547550
kwargs: optional keyword args to be passed to construct the logger.
548551
549552
Returns:
550-
:class:`~ignite.contrib.handlers.clearml_logger.ClearMLLogger`
553+
:class:`~ignite.handlers.clearml_logger.ClearMLLogger`
551554
"""
552555
logger = ClearMLLogger(**kwargs)
553556
_setup_logging(logger, trainer, optimizers, evaluators, log_every_iters)
Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,8 @@
1-
from ignite.contrib.handlers.clearml_logger import ClearMLLogger
2-
from ignite.contrib.handlers.mlflow_logger import MLflowLogger
3-
from ignite.contrib.handlers.neptune_logger import NeptuneLogger
4-
from ignite.contrib.handlers.polyaxon_logger import PolyaxonLogger
5-
from ignite.contrib.handlers.tensorboard_logger import TensorboardLogger
6-
from ignite.contrib.handlers.tqdm_logger import ProgressBar
7-
8-
from ignite.contrib.handlers.visdom_logger import VisdomLogger
9-
from ignite.contrib.handlers.wandb_logger import WandBLogger
101
from ignite.handlers import EpochOutputStore, global_step_from_engine # ref # ref
2+
from ignite.handlers.clearml_logger import ClearMLLogger
113
from ignite.handlers.lr_finder import FastaiLRFinder
4+
from ignite.handlers.mlflow_logger import MLflowLogger
5+
from ignite.handlers.neptune_logger import NeptuneLogger
126
from ignite.handlers.param_scheduler import (
137
ConcatScheduler,
148
CosineAnnealingScheduler,
@@ -18,4 +12,10 @@
1812
ParamGroupScheduler,
1913
PiecewiseLinear,
2014
)
15+
from ignite.handlers.polyaxon_logger import PolyaxonLogger
16+
from ignite.handlers.tensorboard_logger import TensorboardLogger
2117
from ignite.handlers.time_profilers import BasicTimeProfiler, HandlersTimeProfiler
18+
from ignite.handlers.tqdm_logger import ProgressBar
19+
20+
from ignite.handlers.visdom_logger import VisdomLogger
21+
from ignite.handlers.wandb_logger import WandBLogger

0 commit comments

Comments
 (0)