mmpretrain/mmcls/models/heads
Ezra-Yu 0f30c392a3 from core to engine 2022-07-18 11:11:13 +08:00
..
__init__.py Add multi_label heads 2022-07-18 11:11:13 +08:00
base_head.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
cls_head.py from core to engine 2022-07-18 11:11:13 +08:00
conformer_head.py from core to engine 2022-07-18 11:11:13 +08:00
deit_head.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
linear_head.py [Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead 2022-07-18 11:11:13 +08:00
multi_label_cls_head.py from core to engine 2022-07-18 11:11:13 +08:00
multi_label_linear_head.py Add multi_label heads 2022-07-18 11:11:13 +08:00
stacked_head.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
vision_transformer_head.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00