mmpretrain/tests/test_models
yingfhu 62b046521e [Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead 2022-07-18 11:11:13 +08:00
..
test_backbones [Improve] Clean useless code and reduce unit tests memory usage. 2022-07-18 11:11:13 +08:00
test_utils Imporve according to comments 2022-07-18 11:11:13 +08:00
test_classifiers.py [Refactor] Refactor batch augmentations 2022-07-18 11:11:13 +08:00
test_heads.py [Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead 2022-07-18 11:11:13 +08:00
test_neck.py [Feature] Support gem pooling (#677) 2022-02-17 02:17:36 +08:00