30 lines
1.4 KiB
Python
30 lines
1.4 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .checkpoint import CheckpointHook
|
|
from .closure import ClosureHook
|
|
from .ema import EMAHook
|
|
from .evaluation import DistEvalHook, EvalHook
|
|
from .hook import HOOKS, Hook
|
|
from .iter_timer import IterTimerHook
|
|
from .logger import (DvcliveLoggerHook, LoggerHook, MlflowLoggerHook,
|
|
NeptuneLoggerHook, PaviLoggerHook, TensorboardLoggerHook,
|
|
TextLoggerHook, WandbLoggerHook)
|
|
from .lr_updater import LrUpdaterHook
|
|
from .memory import EmptyCacheHook
|
|
from .momentum_updater import MomentumUpdaterHook
|
|
from .optimizer import (Fp16OptimizerHook, GradientCumulativeFp16OptimizerHook,
|
|
GradientCumulativeOptimizerHook, OptimizerHook)
|
|
from .profiler import ProfilerHook
|
|
from .sampler_seed import DistSamplerSeedHook
|
|
from .sync_buffer import SyncBuffersHook
|
|
|
|
__all__ = [
|
|
'HOOKS', 'Hook', 'CheckpointHook', 'ClosureHook', 'LrUpdaterHook',
|
|
'OptimizerHook', 'Fp16OptimizerHook', 'IterTimerHook',
|
|
'DistSamplerSeedHook', 'EmptyCacheHook', 'LoggerHook', 'MlflowLoggerHook',
|
|
'PaviLoggerHook', 'TextLoggerHook', 'TensorboardLoggerHook',
|
|
'NeptuneLoggerHook', 'WandbLoggerHook', 'DvcliveLoggerHook',
|
|
'MomentumUpdaterHook', 'SyncBuffersHook', 'EMAHook', 'EvalHook',
|
|
'DistEvalHook', 'ProfilerHook', 'GradientCumulativeOptimizerHook',
|
|
'GradientCumulativeFp16OptimizerHook'
|
|
]
|