mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
a9b6753fbe
* Add build_runner_with_tta and PrepareTTAHook * rename hook file * support build tta runner with runner type * add unit test * Add build_runner_with_tta to index.rst * minor refine * Add runner test cast * Fix unit test * fix unit test * tmp save * pop None if key does not exist * Fix is_model_wrapper and force register class in test_runner * [Fix] Fix is_model_wrapper * destroy group after ut * register module in testcase * pass through unit test * fix as comment * remove breakpoint * remove mmengine/testing/runner_test_cast.py * minor refine * minor refine * minor refine * set default data preprocessor for model * minor refine * minor refine Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * fix lint * Fix unit test * replace with in ImgDataPreprocessor * Fix as comment * add inference tutorial in advanced tutorial * update index.rst * add tta example * refine tta tutorial * Add english tutorial * add note for build_runner_with_tta * Fix as comment * add examples * remove chinese comment * Update docs/en/advanced_tutorials/test_time_augmentation.md Co-authored-by: RangiLyu <lyuchqi@gmail.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by: RangiLyu <lyuchqi@gmail.com>
22 lines
873 B
Python
22 lines
873 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .checkpoint_hook import CheckpointHook
|
|
from .ema_hook import EMAHook
|
|
from .empty_cache_hook import EmptyCacheHook
|
|
from .hook import Hook
|
|
from .iter_timer_hook import IterTimerHook
|
|
from .logger_hook import LoggerHook
|
|
from .naive_visualization_hook import NaiveVisualizationHook
|
|
from .param_scheduler_hook import ParamSchedulerHook
|
|
from .profiler_hook import ProfilerHook
|
|
from .runtime_info_hook import RuntimeInfoHook
|
|
from .sampler_seed_hook import DistSamplerSeedHook
|
|
from .sync_buffer_hook import SyncBuffersHook
|
|
from .test_time_aug_hook import PrepareTTAHook
|
|
|
|
__all__ = [
|
|
'Hook', 'IterTimerHook', 'DistSamplerSeedHook', 'ParamSchedulerHook',
|
|
'SyncBuffersHook', 'EmptyCacheHook', 'CheckpointHook', 'LoggerHook',
|
|
'NaiveVisualizationHook', 'EMAHook', 'RuntimeInfoHook', 'ProfilerHook',
|
|
'PrepareTTAHook'
|
|
]
|