mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmclassification/mmcls/models/heads
History
Ma Zerun 20f9ace5c5
[CI] Add Circle CI for mmcls 1.x. (#970) 
* [CI] Add Circle CI for mmcls 1.x

* Fix circle-ci config

* Use `torch.cat` instead of `torch.hstack` to compat with PyTorch 1.6

* Compat the behavior of LongTensor in PyTorch 1.6

* Avoid random behavior in the unit test of Lighting

* Fix cuda ci

* Remove github workflow temporarily.
 		2022-08-22 15:02:08 +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 [CI] Add Circle CI for mmcls 1.x. (#970) 2022-08-22 15:02:08 +08:00
conformer_head.py [CI] Add Circle CI for mmcls 1.x. (#970) 2022-08-22 15:02:08 +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 [Improve] Rename `mmcls.data` to `mmcls.structures`. (#941) 2022-07-29 14:18: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