mmpretrain

Commit Graph

Author	SHA1	Message	Date
takuoko	841256b630	[Feature] Support RetrieverRecall metric & Add ArcFace config (#1316 ) * rebase * add ap metric * fix mlti-gpu bug in retrevel * rebase * rebase * add training cfgs and update readme.md * fix bugs(cannot load vecs in dist and diff test-val recall\) * update configs and readme * fix ut * fix doc * rebase * fix rebase conflicts * fix rebase error * fix UT error * fix docs * fix typo --------- Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>	2023-02-14 12:46:21 +08:00
Ezra-Yu	705ed2be49	[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
Ma Zerun	c127c474b9	[Feature] Support getting model from the name defined in the model-index file. (#1236 ) * [Feature] Support getting model from the name defined in the model-index file. * Add unit tests. * Prevent import `timm` if the `TIMMBackbone` is not used. * Fix Windows CI. * Move `init_model` to `mmcls.apis.hub`, and support pass nn.Module to all model components. * Fix requirements * Rename `hub.py` to `model.py` and add unit tests.	2022-12-06 17:00:22 +08:00
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

4 Commits (841256b630f777bc0bc76ed0bba776ec341e9204)