mmpretrain/mmcls/models
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
..
backbones [Feature] Support LeViT backbone. (#1238) 2023-01-17 17:43:42 +08:00
classifiers [Feature] Support getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +08:00
heads [Feature] Support LeViT backbone. (#1238) 2023-01-17 17:43:42 +08:00
losses [Fix] Fix `loss_weight` in `LabelSmoothLoss`. (#1058) 2022-09-28 08:18:12 +08:00
necks [Enhance] Enable to toggle whether Gem Pooling is trainable or not. (#1246) 2023-02-09 11:27:05 +08:00
retrievers [Feature] Support RetrieverRecall metric & Add ArcFace config (#1316) 2023-02-14 12:46:21 +08:00
tta [Feature] Support TTA and add `--tta` in `tools/test.py`. (#1161) 2022-12-30 11:46:17 +08:00
utils [Feature] Support ConvNeXt-V2 backbone. (#1294) 2023-01-06 16:13:41 +08:00
__init__.py [Feature] Support TTA and add `--tta` in `tools/test.py`. (#1161) 2022-12-30 11:46:17 +08:00
builder.py [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00