a b/mmseg/utils/__init__.py
1
# Copyright (c) OpenMMLab. All rights reserved.
2
from .collect_env import collect_env
3
from .logger import get_root_logger
4
from .checkpoint import load_checkpoint
5
from .optimizer import (ApexOptimizerHook, GradientCumulativeOptimizerHook, GradientCumulativeFp16OptimizerHook)
6
from .customized_text import CustomizedTextLoggerHook
7
from .layer_decay_optimizer_constructor import LearningRateDecayOptimizerConstructor
8
9
__all__ = ['get_root_logger', 'collect_env', 'load_checkpoint', 'ApexOptimizerHook', 'GradientCumulativeOptimizerHook', 'GradientCumulativeFp16OptimizerHook', 'CustomizedTextLoggerHook', 'LearningRateDecayOptimizerConstructor']