mmclassification/mmpretrain/models/heads
Yixiao Fang 63d9f27fde
[Refactor] Add necks, heads and losses for the self-supervised task. (#1376)
* add necks

* refactor linear neck

* rename simmim neck

* add heads

* add losses

* fix

* add unittest

* update

* update cae

* remove mim head

* update config
2023-02-28 10:05:00 +08:00
..
__init__.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
base_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
beitv1_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
beitv2_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
cae_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
cls_head.py [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
conformer_head.py [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
contrastive_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
deit_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
efficientformer_head.py [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
latent_heads.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
levit_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
linear_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
mae_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
margin_head.py [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
maskfeat_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
mixmim_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
mocov3_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
multi_label_cls_head.py [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
multi_label_csra_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
multi_label_linear_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
multi_task_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
simmim_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
stacked_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
swav_head.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
vig_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
vision_transformer_head.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00