mmclassification/mmpretrain/models/necks
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
beitv2_neck.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
cae_neck.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
densecl_neck.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
gap.py
gem.py
hr_fuse.py
linear_neck.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
mae_neck.py [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
milan_neck.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
mixmim_neck.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
mocov2_neck.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
nonlinear_neck.py [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
simmim_neck.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
swav_neck.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00