|
15 | 15 | from torch.optim.lr_scheduler import _LRScheduler as PyTorchLRScheduler |
16 | 16 |
|
17 | 17 | 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, |
19 | 22 | ClearMLLogger, |
| 23 | + DiskSaver, |
| 24 | + EarlyStopping, |
20 | 25 | global_step_from_engine, |
21 | 26 | MLflowLogger, |
22 | 27 | NeptuneLogger, |
23 | 28 | PolyaxonLogger, |
24 | 29 | ProgressBar, |
25 | 30 | TensorboardLogger, |
| 31 | + TerminateOnNan, |
26 | 32 | VisdomLogger, |
27 | 33 | WandBLogger, |
28 | 34 | ) |
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 |
33 | 36 | from ignite.handlers.checkpoint import BaseSaveHandler |
34 | 37 | from ignite.handlers.param_scheduler import ParamScheduler |
35 | 38 | from ignite.metrics import RunningAverage |
@@ -361,7 +364,7 @@ def setup_tb_logging( |
361 | 364 | kwargs: optional keyword args to be passed to construct the logger. |
362 | 365 |
|
363 | 366 | Returns: |
364 | | - :class:`~ignite.contrib.handlers.tensorboard_logger.TensorboardLogger` |
| 367 | + :class:`~ignite.handlers.tensorboard_logger.TensorboardLogger` |
365 | 368 | """ |
366 | 369 | logger = TensorboardLogger(log_dir=output_path, **kwargs) |
367 | 370 | _setup_logging(logger, trainer, optimizers, evaluators, log_every_iters) |
@@ -392,7 +395,7 @@ def setup_visdom_logging( |
392 | 395 | kwargs: optional keyword args to be passed to construct the logger. |
393 | 396 |
|
394 | 397 | Returns: |
395 | | - :class:`~ignite.contrib.handlers.visdom_logger.VisdomLogger` |
| 398 | + :class:`~ignite.handlers.visdom_logger.VisdomLogger` |
396 | 399 | """ |
397 | 400 | logger = VisdomLogger(**kwargs) |
398 | 401 | _setup_logging(logger, trainer, optimizers, evaluators, log_every_iters) |
@@ -423,7 +426,7 @@ def setup_mlflow_logging( |
423 | 426 | kwargs: optional keyword args to be passed to construct the logger. |
424 | 427 |
|
425 | 428 | Returns: |
426 | | - :class:`~ignite.contrib.handlers.mlflow_logger.MLflowLogger` |
| 429 | + :class:`~ignite.handlers.mlflow_logger.MLflowLogger` |
427 | 430 | """ |
428 | 431 | logger = MLflowLogger(**kwargs) |
429 | 432 | _setup_logging(logger, trainer, optimizers, evaluators, log_every_iters) |
@@ -454,7 +457,7 @@ def setup_neptune_logging( |
454 | 457 | kwargs: optional keyword args to be passed to construct the logger. |
455 | 458 |
|
456 | 459 | Returns: |
457 | | - :class:`~ignite.contrib.handlers.neptune_logger.NeptuneLogger` |
| 460 | + :class:`~ignite.handlers.neptune_logger.NeptuneLogger` |
458 | 461 | """ |
459 | 462 | logger = NeptuneLogger(**kwargs) |
460 | 463 | _setup_logging(logger, trainer, optimizers, evaluators, log_every_iters) |
@@ -485,7 +488,7 @@ def setup_wandb_logging( |
485 | 488 | kwargs: optional keyword args to be passed to construct the logger. |
486 | 489 |
|
487 | 490 | Returns: |
488 | | - :class:`~ignite.contrib.handlers.wandb_logger.WandBLogger` |
| 491 | + :class:`~ignite.handlers.wandb_logger.WandBLogger` |
489 | 492 | """ |
490 | 493 | logger = WandBLogger(**kwargs) |
491 | 494 | _setup_logging(logger, trainer, optimizers, evaluators, log_every_iters) |
@@ -516,7 +519,7 @@ def setup_plx_logging( |
516 | 519 | kwargs: optional keyword args to be passed to construct the logger. |
517 | 520 |
|
518 | 521 | Returns: |
519 | | - :class:`~ignite.contrib.handlers.polyaxon_logger.PolyaxonLogger` |
| 522 | + :class:`~ignite.handlers.polyaxon_logger.PolyaxonLogger` |
520 | 523 | """ |
521 | 524 | logger = PolyaxonLogger(**kwargs) |
522 | 525 | _setup_logging(logger, trainer, optimizers, evaluators, log_every_iters) |
@@ -547,7 +550,7 @@ def setup_clearml_logging( |
547 | 550 | kwargs: optional keyword args to be passed to construct the logger. |
548 | 551 |
|
549 | 552 | Returns: |
550 | | - :class:`~ignite.contrib.handlers.clearml_logger.ClearMLLogger` |
| 553 | + :class:`~ignite.handlers.clearml_logger.ClearMLLogger` |
551 | 554 | """ |
552 | 555 | logger = ClearMLLogger(**kwargs) |
553 | 556 | _setup_logging(logger, trainer, optimizers, evaluators, log_every_iters) |
|
0 commit comments