mmclassification/mmpretrain/utils/progress.py

# Copyright (c) OpenMMLab. All rights reserved.
import mmengine.dist as dist
import rich.progress as progress


def track_on_main_process(sequence, *args, **kwargs):
    if not dist.is_main_process():
        yield from sequence
    else:
        yield from progress.track(sequence, *args, **kwargs)
[Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) * 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> 2022-11-02 17:43:56 +08:00			`# Copyright (c) OpenMMLab. All rights reserved.`
			`import mmengine.dist as dist`
			`import rich.progress as progress`


			`def track_on_main_process(sequence, args, *kwargs):`
			`if not dist.is_main_process():`
[Fix] Fix retrieval multi gpu bug (#1319) * fix mlti-gpu bug in retrevel * fix bugs(cannot load vecs in dist and diff test-val recall\) * load weight each process 2023-02-09 15:55:47 +08:00			`yield from sequence`
			`else:`
			`yield from progress.track(sequence, args, *kwargs)`