Ma Zerun dbf3df21a3

[Refactor] Use out_type to specify ViT-like backbone output. (#1408) ...

* [Refactor] Use  to specify ViT-like backbone output.

* Fix ClsBatchNormNeck

* Update mmpretrain/models/necks/mae_neck.py

---------

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>

2023-03-09 11:02:58 +08:00

..

__init__.py

[Refactor] Add selfsup algorithms. (#1389)

2023-03-06 16:53:15 +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

[Refactor] Refactor ClsDatasample to a union DataSample. (#1371)

2023-02-23 10:07:53 +08:00

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

[Refactor] Refactor ClsDatasample to a union DataSample. (#1371)

2023-02-23 10:07:53 +08:00

latent_heads.py

[Refactor] Add selfsup algorithms. (#1389)

2023-03-06 16:53:15 +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 selfsup algorithms. (#1389)

2023-03-06 16:53:15 +08:00

margin_head.py

[Refactor] Refactor ClsDatasample to a union DataSample. (#1371)

2023-02-23 10:07:53 +08:00

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

[Refactor] Add selfsup algorithms. (#1389)

2023-03-06 16:53:15 +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

[Refactor] Add selfsup algorithms. (#1389)

2023-03-06 16:53:15 +08:00

simmim_head.py

[Refactor] Add selfsup algorithms. (#1389)

2023-03-06 16:53:15 +08:00

stacked_head.py

Rename the package name to mmpretrain.

2023-02-17 15:20:55 +08:00

swav_head.py

[Refactor] Add selfsup algorithms. (#1389)

2023-03-06 16:53:15 +08:00

vig_head.py

Rename the package name to mmpretrain.

2023-02-17 15:20:55 +08:00

vision_transformer_head.py

[Refactor] Use out_type to specify ViT-like backbone output. (#1408)

2023-03-09 11:02:58 +08:00