|
|
|
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' |
|
] |
|
|