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

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 necks, heads and losses for the self-supervised task. (#1376)

2023-02-28 10:05:00 +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

Rename the package name to mmpretrain.

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

gem.py

Rename the package name to mmpretrain.

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

hr_fuse.py

Rename the package name to mmpretrain.

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

linear_neck.py

[Refactor] Add necks, heads and losses for the self-supervised task. (#1376)

2023-02-28 10:05:00 +08:00

mae_neck.py

[Refactor] Add necks, heads and losses for the self-supervised task. (#1376)

2023-02-28 10:05:00 +08:00

milan_neck.py

[Refactor] Add necks, heads and losses for the self-supervised task. (#1376)

2023-02-28 10:05:00 +08:00

mixmim_neck.py

[Refactor] Add necks, heads and losses for the self-supervised task. (#1376)

2023-02-28 10:05:00 +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] Add necks, heads and losses for the self-supervised task. (#1376)

2023-02-28 10:05:00 +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