mmpretrain/mmcls/models
fam_taro 4ce7be17c9
[Enhance] Enable to toggle whether Gem Pooling is trainable or not. (#1246)
* Enable to toggle whether Gem Pooling is trainable or not.

* Add test case whether Gem Pooling is trainable or not.

* Enable to toggle whether Gem Pooling is trainable or not by requires_grad

---------

Co-authored-by: Yusuke Fujimoto <yusuke.fujimoto@rist.co.jp>
2023-02-09 11:27:05 +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 getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +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