zzc98 693596bc2f

[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

..

__init__.py

[Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055)

2022-11-02 17:43:56 +08:00

collect_env.py

[Refactor] Refactor the data flow. (#989)

2022-08-26 10:40:43 +08:00

progress.py

[Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055)

2022-11-02 17:43:56 +08:00

setup_env.py

[Improve] Multiple fixes. (#957)

2022-08-09 15:58:08 +08:00