mmclassification/mmpretrain/models/heads
Yixiao Fang a1cfe888e2
[Feature] Support SparK. (#1531)
* add spark configs

* fix configs

* remove repeat aug

* add module codes

* support lr layer decay of resnet

* update

* fix lint

* add metafile and readme

* fix lint

* add models and logs

* refactor codes

* fix lint

* update model rst

* update name

* add docstring

* add ut

* fix lint

---------

Co-authored-by: Ma Zerun <mzr1996@163.com>
2023-06-19 11:27:50 +08:00
..
__init__.py [Feature] Support SparK. (#1531) 2023-06-19 11:27:50 +08:00
beitv1_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
beitv2_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
cae_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
cls_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
conformer_head.py
contrastive_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
deit_head.py [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
efficientformer_head.py
grounding_head.py [Feature] Support multiple multi-modal algorithms and inferencers. (#1561) 2023-05-19 16:50:04 +08:00
itc_head.py [Feature] Support multiple multi-modal algorithms and inferencers. (#1561) 2023-05-19 16:50:04 +08:00
itm_head.py [Feature] Support multiple multi-modal algorithms and inferencers. (#1561) 2023-05-19 16:50:04 +08:00
itpn_clip_head.py [Feature] Support iTPN and HiViT (#1584) 2023-05-26 12:08:34 +08:00
latent_heads.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
levit_head.py
linear_head.py
mae_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
margin_head.py
mim_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
mixmim_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
mocov3_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
multi_label_cls_head.py [Fix] Fix the bug in binary cross entropy loss (#1499) 2023-04-19 13:53:31 +08:00
multi_label_csra_head.py
multi_label_linear_head.py
multi_task_head.py [Fix] Fix multi-task-head loss potential bug (#1530) 2023-05-06 18:04:34 +08:00
seq_gen_head.py [Feature] Support multiple multi-modal algorithms and inferencers. (#1561) 2023-05-19 16:50:04 +08:00
simmim_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
spark_head.py [Feature] Support SparK. (#1531) 2023-06-19 11:27:50 +08:00
stacked_head.py
swav_head.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
vig_head.py
vision_transformer_head.py [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
vqa_head.py [Feature] Support multiple multi-modal algorithms and inferencers. (#1561) 2023-05-19 16:50:04 +08:00