mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
693596bc2f
* feat: add image retriever * feat: add image retriever * feat: add image retriever * feat: add image retriever * feat: add image retriever * feat: add image retriever * feat: add image retriever * feat: add image retriever * feat: add image retriever * update retriever * fix lint * add hook unit test * Use `register_buffer` to save prototype vectors and add a progress bar during preparing prototype. * update UTs * update UTs * fix typo * modify the hook Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> Co-authored-by: mzr1996 <mzr1996@163.com>
7 lines
249 B
Python
7 lines
249 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .collect_env import collect_env
|
|
from .progress import track_on_main_process
|
|
from .setup_env import register_all_modules
|
|
|
|
__all__ = ['collect_env', 'register_all_modules', 'track_on_main_process']
|