--- a +++ b/mmseg/utils/__init__.py @@ -0,0 +1,9 @@ +# Copyright (c) OpenMMLab. All rights reserved. +from .collect_env import collect_env +from .logger import get_root_logger +from .checkpoint import load_checkpoint +from .optimizer import (ApexOptimizerHook, GradientCumulativeOptimizerHook, GradientCumulativeFp16OptimizerHook) +from .customized_text import CustomizedTextLoggerHook +from .layer_decay_optimizer_constructor import LearningRateDecayOptimizerConstructor + +__all__ = ['get_root_logger', 'collect_env', 'load_checkpoint', 'ApexOptimizerHook', 'GradientCumulativeOptimizerHook', 'GradientCumulativeFp16OptimizerHook', 'CustomizedTextLoggerHook', 'LearningRateDecayOptimizerConstructor']