mmpretrain/mmcls/models/heads
Ezra-Yu 34d5a25281
[Refactor] Support passing arguments to loss from head. (#523)
2021-11-10 17:12:34 +08:00
..
__init__.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
base_head.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
cls_head.py [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
linear_head.py [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
multi_label_head.py [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
multi_label_linear_head.py [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
stacked_head.py [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
vision_transformer_head.py [Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00