mmclassification/mmpretrain/models/heads
Ezra-Yu 7f4eccbecf
[Fix] Fix multi-task-head loss potential bug (#1530)
* fix bug

* add comments
2023-05-06 18:04:34 +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 [Fix] Fix the bug in binary cross entropy loss (#1499) 2023-04-19 13:53:31 +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 [Fix] Fix multi-task-head loss potential bug (#1530) 2023-05-06 18:04:34 +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