mmclassification/mmpretrain/models/losses
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
asymmetric_loss.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
cae_loss.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
cosine_similarity_loss.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
cross_correlation_loss.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
cross_entropy_loss.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
focal_loss.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
label_smooth_loss.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
reconstruction_loss.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
seesaw_loss.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
swav_loss.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
utils.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00